mistral

app.py

@@ -75,12 +75,17 @@ css = """
 """
 
 def get_messages_formatter_type(model_name):
+    print(f"getting type for model: {model_name}")
     if "Llama" in model_name:
         return MessagesFormatterType.LLAMA_3
+    elif "Mistral" in model_name:
+        return MessagesFormatterType.MISTRAL
     elif "unsloth" in model_name:
         return MessagesFormatterType.CHATML
     else:
-        raise ValueError(f"Unsupported model: {model_name}")
+        print("formatter type not found, trying default")
+        # raise ValueError(f"Unsupported model: {model_name}")
+        return MessagesFormatterType.CHATML
 
 @spaces.GPU(duration=20)
 def respond(