Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -74,6 +74,8 @@ def get_messages_formatter_type(model_name):
|
|
74 |
return MessagesFormatterType.LLAMA_3
|
75 |
elif "unsloth" in model_name:
|
76 |
return MessagesFormatterType.CHATML
|
|
|
|
|
77 |
else:
|
78 |
raise ValueError(f"Unsupported model: {model_name}")
|
79 |
|
|
|
74 |
return MessagesFormatterType.LLAMA_3
|
75 |
elif "unsloth" in model_name:
|
76 |
return MessagesFormatterType.CHATML
|
77 |
+
elif "Mistral" in model_name: # Mistral 모델 지원 추가
|
78 |
+
return MessagesFormatterType.CHATML # Mistral은 일반적으로 ChatML 형식을 사용합니다
|
79 |
else:
|
80 |
raise ValueError(f"Unsupported model: {model_name}")
|
81 |
|