Spaces:
Sleeping
Sleeping
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 |
|