ginipick commited on
Commit
9ecffc1
·
verified ·
1 Parent(s): 8db507a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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