Ibraaheem commited on
Commit
a06a3a4
·
1 Parent(s): 2054738

Update private_gpt/components/llm/llm_component.py

Browse files
private_gpt/components/llm/llm_component.py CHANGED
@@ -103,6 +103,12 @@ class LLMComponent:
103
  verbose=True,
104
  )
105
  logger.info("Initializing the LLM Model in=%s", "Mistral-7B")
 
 
 
 
 
 
106
 
107
 
108
 
 
103
  verbose=True,
104
  )
105
  logger.info("Initializing the LLM Model in=%s", "Mistral-7B")
106
+
107
+ def switch_to_model(self, new_model: str):
108
+ if self.llm_mode == "dynamic":
109
+ self.switch_model(new_model) # Calls the nested function
110
+ else:
111
+ logger.warning("Model switching not supported in current mode: %s", self.llm_mode)
112
 
113
 
114