Fix bug in get_model_id function

swe-bench · Mar 27, 2024 · 590a050 · 590a050
1 parent caa5e65
commit 590a050
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/inference/run_api.py b/inference/run_api.py
@@ -211,8 +211,10 @@ def get_model_id(model_name: str) -> str:
     """
     if "deepseek" in model_name:
         return "deepseek"
-    if "codellama" in model_name:
+    elif "codellama" in model_name:
         return "codelllma"
+    else:
+        raise ValueError(f"Unknown model name: {model_name}")
 
 
 def claude_tokenize(string: str, api) -> int: