Add new model version for gpt-3.5-turbo-finetuned

sotopia-lab · Jan 16, 2024 · e4f531f · e4f531f
1 parent 6cc9c6d
commit e4f531f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sotopia/generation_utils/generate.py b/sotopia/generation_utils/generate.py
@@ -50,6 +50,7 @@
     "togethercomputer/llama-2-70b-chat",
     "togethercomputer/mpt-30b-chat",
     "gpt-3.5-turbo",
+    "gpt-3.5-turbo-finetuned",
     "text-davinci-003",
     "gpt-4",
     "gpt-4-turbo",
@@ -294,6 +295,7 @@ def _return_fixed_model_version(
 ) -> str:
     return {
         "gpt-3.5-turbo": "gpt-3.5-turbo-0613",
+        "gpt-3.5-turbo-finetuned": "ft:gpt-3.5-turbo-0613:academicscmu::8hK7f26o",
         "gpt-4": "gpt-4-0613",
         "gpt-4-turbo": "gpt-4-1106-preview",
     }[model_name]