CineAI commited on
Commit
230ae1d
1 Parent(s): 0871e09
Files changed (1) hide show
  1. 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, HF_Mistaril, HF_TinyLlama, HF_SmolLM135, HF_SmolLM360, HF_SmolLM, HF_Gemma2, HF_Qwen2].
94
  """
95
 
96
- self.model_classes = [Gemini, HF_Mistaril, HF_TinyLlama, HF_SmolLM135, HF_SmolLM360, HF_SmolLM, HF_Gemma2, HF_Qwen2]
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: