added support to phi3 model

Deeplearn-PeD · May 4, 2024 · 46b9ff9 · 46b9ff9
1 parent 6eb7fc0
commit 46b9ff9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/base_agent/llminterface.py b/base_agent/llminterface.py
@@ -30,6 +30,9 @@ def get_response(self, question: str, context: str = None) -> str:
         elif 'wizard' in self.model:
             self.model = 'wizardlm2'
             return self.get_ollama_response(question, context)
+        elif 'phi' in self.model:
+            self.model = 'phi3'
+            return self.get_ollama_response(question, context)
 
     def get_gpt_response(self, question: str, context: str)->str:
         response = self.llm.chat.completions.create(