gpt-4-turbo in azure

GooeyAI · Nov 20, 2023 · 2e980c3 · 2e980c3
1 parent 493ceb9
commit 2e980c3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/daras_ai_v2/language_model.py b/daras_ai_v2/language_model.py
@@ -78,7 +78,10 @@ def is_chat_model(self) -> bool:
 AZURE_OPENAI_MODEL_PREFIX = "openai-"
 
 llm_model_names = {
-    LargeLanguageModels.gpt_4_turbo: "gpt-4-1106-preview",
+    LargeLanguageModels.gpt_4_turbo: (
+        "openai-gpt-4-turbo-prod-ca-1",
+        "gpt-4-1106-preview",
+    ),
     LargeLanguageModels.gpt_4: (
         "openai-gpt-4-prod-ca-1",
         "gpt-4",