bug fix for return fixed model version function

sotopia-lab · Oct 1, 2024 · db8839e · db8839e
1 parent 1923c18
commit db8839e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sotopia/generation_utils/generate.py b/sotopia/generation_utils/generate.py
@@ -281,7 +281,7 @@ def _type(self) -> str:
 
 def _return_fixed_model_version(model_name: str) -> str:
     if model_name in [
-        "gpt-4o-mini",
+        "gpt-3.5-turbo",
         "gpt-3.5-turbo-finetuned",
         "gpt-3.5-turbo-ft-MF",
         "gpt-4",