From 342ba0288b3a0bbb0f4b16af0f854720d3634a90 Mon Sep 17 00:00:00 2001 From: Soumik Rakshit <19soumik.rakshit96@gmail.com> Date: Wed, 30 Oct 2024 23:15:13 +0000 Subject: [PATCH] update: tests --- hemm/tests/test_prompt_alignment_eval.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hemm/tests/test_prompt_alignment_eval.py b/hemm/tests/test_prompt_alignment_eval.py index 217e549..d0744ac 100644 --- a/hemm/tests/test_prompt_alignment_eval.py +++ b/hemm/tests/test_prompt_alignment_eval.py @@ -13,9 +13,9 @@ def test_prompt_alignment_evaluation(): enable_cpu_offfload=False, ) clip_scorer = CLIPScoreMetric( - clip_model_name_or_path="openai/clip-vit-base-patch16" + model_name="openai/clip-vit-base-patch16" ) - clip_iqa_scorer = CLIPImageQualityScoreMetric(clip_model_name_or_path="clip_iqa") + clip_iqa_scorer = CLIPImageQualityScoreMetric(model_name="clip_iqa") dataset = weave.ref("parti-prompts:v0").get().rows[:2] evaluation = weave.Evaluation( dataset=dataset, scorers=[clip_scorer, clip_iqa_scorer]