From 4f29ad02e1da166f9819df246b62ad9fe161ae11 Mon Sep 17 00:00:00 2001 From: Soumik Rakshit <19soumik.rakshit96@gmail.com> Date: Wed, 30 Oct 2024 23:24:10 +0000 Subject: [PATCH] fix: lint --- hemm/tests/test_prompt_alignment_eval.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/hemm/tests/test_prompt_alignment_eval.py b/hemm/tests/test_prompt_alignment_eval.py index d0744ac..bb9b0ca 100644 --- a/hemm/tests/test_prompt_alignment_eval.py +++ b/hemm/tests/test_prompt_alignment_eval.py @@ -12,9 +12,7 @@ def test_prompt_alignment_evaluation(): diffusion_model_name_or_path="CompVis/stable-diffusion-v1-4", enable_cpu_offfload=False, ) - clip_scorer = CLIPScoreMetric( - model_name="openai/clip-vit-base-patch16" - ) + clip_scorer = CLIPScoreMetric(model_name="openai/clip-vit-base-patch16") clip_iqa_scorer = CLIPImageQualityScoreMetric(model_name="clip_iqa") dataset = weave.ref("parti-prompts:v0").get().rows[:2] evaluation = weave.Evaluation(