Add support for gpt-4-turbo

HumanSignal · Nov 7, 2023 · 8ac8cc9 · 8ac8cc9
1 parent 8c722d5
commit 8ac8cc9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/adala/runtimes/openai.py b/adala/runtimes/openai.py
@@ -55,7 +55,7 @@ def init_runtime(self):
         self._check_model_availability()
 
         student_models = {'gpt-3.5-turbo-instruct', 'text-davinci-003'}
-        teacher_models = {'gpt-4', 'gpt-3.5-turbo', 'gpt-3.5-turbo-16k'}
+        teacher_models = {'gpt-4', 'gpt-3.5-turbo', 'gpt-3.5-turbo-16k', 'gpt-4-1106-preview', 'gpt-4-vision-preview'}
 
         if self.gpt_model_name in student_models:
             self.llm_runtime_type = LLMRuntimeType.STUDENT