Bug fix

knoriy · Dec 13, 2023 · faec8a5 · faec8a5
1 parent 86d3f1e
commit faec8a5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/clara/eval/test_zeroshot.py b/clara/eval/test_zeroshot.py
@@ -108,7 +108,7 @@ def main(args):
 	##############
 	dataset.setup()
 
-	zeroshot_weights, all_texts = zeroshot_classifier(model, classes, templates)
+	zeroshot_weights, all_texts = zeroshot_text(model, classes, templates)
 	tops = run(model, zeroshot_weights, dataset.test_dataloader(), metric, args.task)
 
 	return tops
@@ -130,4 +130,4 @@ def main(args):
 
 	tops = main(args)
 
-	pprint(tops)
+	pprint(tops)