Fixed the issue with CLI option

openvinotoolkit · Oct 29, 2024 · 72a8627 · 72a8627
1 parent 565cb6c
commit 72a8627
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/who_what_benchmark/whowhatbench/wwb.py b/tools/who_what_benchmark/whowhatbench/wwb.py
@@ -484,7 +484,7 @@ def create_evaluator(base_model, args):
         EvaluatorCLS = EVALUATOR_REGISTRY[task]
         prompts = load_prompts(args)
 
-        if task == "text-generation":
+        if task == "text":
             tokenizer = load_tokenizer(args)
             return EvaluatorCLS(
                 base_model=base_model,