From 8ac8cc9f190471d18c79d4b7976e282247d4f8b0 Mon Sep 17 00:00:00 2001
From: nik <nik@heartex.net>
Date: Tue, 7 Nov 2023 17:30:17 +0000
Subject: [PATCH] Add support for gpt-4-turbo

---
 adala/runtimes/openai.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/adala/runtimes/openai.py b/adala/runtimes/openai.py
index 2af9d70b..7e0ea60c 100644
--- 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