wanzin commited on
Commit
9eac0b7
Β·
1 Parent(s): 1a9bde9

fixed the evaluator

Browse files
Files changed (1) hide show
  1. 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":