Correctly parse model names for starcoder-2-15b (#398)

starcoder-2-15b's model name incorrectly parsed as gpt- due to get use_case from model config.
openvinotoolkit · Apr 29, 2024 · e9825f2 · e9825f2
1 parent 7365730
commit e9825f2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/llm_bench/python/utils/config_class.py b/llm_bench/python/utils/config_class.py
@@ -101,7 +101,8 @@
         "deci",
         "internlm",
         "olmo",
-        "phi3"
+        "phi3",
+        "starcoder"
     ],
     'ldm_super_resolution': ['ldm-super-resolution'],
 }