fixed the evaluator
Browse files- clip_eval.py +1 -1
clip_eval.py
CHANGED
@@ -50,7 +50,7 @@ class DmxClipEval(evaluate.Metric):
|
|
50 |
)
|
51 |
|
52 |
def clip_dataset_evaluator(
|
53 |
-
self, model, device, dataset_name="mscoco", n_examples=-1
|
54 |
):
|
55 |
processor = CLIPProcessor.from_pretrained(model.config._name_or_path)
|
56 |
if dataset_name == "mscoco":
|
|
|
50 |
)
|
51 |
|
52 |
def clip_dataset_evaluator(
|
53 |
+
self, model, device, desc, dataset_name="mscoco", n_examples=-1
|
54 |
):
|
55 |
processor = CLIPProcessor.from_pretrained(model.config._name_or_path)
|
56 |
if dataset_name == "mscoco":
|