temporary
Browse files- llm/utils/chat.py +2 -2
llm/utils/chat.py
CHANGED
@@ -90,10 +90,10 @@ class Conversation:
|
|
90 |
|
91 |
Args:
|
92 |
model_classes (list, optional): A list of LLM model classes to try in sequence.
|
93 |
-
Defaults to [Gemini,
|
94 |
"""
|
95 |
|
96 |
-
self.model_classes = [
|
97 |
self.current_model_index = 0
|
98 |
|
99 |
def _get_conversation(self) -> Any:
|
|
|
90 |
|
91 |
Args:
|
92 |
model_classes (list, optional): A list of LLM model classes to try in sequence.
|
93 |
+
Defaults to [Gemini, HF_SmolLM135, HF_SmolLM360, HF_TinyLlama, HF_SmolLM, HF_Gemma2, HF_Mistaril, HF_Qwen2].
|
94 |
"""
|
95 |
|
96 |
+
self.model_classes = [HF_SmolLM135, HF_SmolLM360, HF_TinyLlama, HF_SmolLM, HF_Gemma2, HF_Mistaril, HF_Qwen2]
|
97 |
self.current_model_index = 0
|
98 |
|
99 |
def _get_conversation(self) -> Any:
|