From e9825f2a9bd72261586c554099f36de35067419b Mon Sep 17 00:00:00 2001
From: guozhong wang <guozhong.wang@intel.com>
Date: Mon, 29 Apr 2024 12:45:49 +0800
Subject: [PATCH] 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.
---
 llm_bench/python/utils/config_class.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/llm_bench/python/utils/config_class.py b/llm_bench/python/utils/config_class.py
index 47fb4050e8..ac50158f41 100644
--- 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'],
 }