ginipick commited on
Commit
01b053a
ยท
verified ยท
1 Parent(s): 9ecffc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -74,8 +74,10 @@ def get_messages_formatter_type(model_name):
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
 
 
74
  return MessagesFormatterType.LLAMA_3
75
  elif "unsloth" in model_name:
76
  return MessagesFormatterType.CHATML
77
+ elif "Mistral" in model_name:
78
+ # Mistral ๋ชจ๋ธ์€ ๊ธฐ๋ณธ์ ์œผ๋กœ ํŠน์ • ํ…œํ”Œ๋ฆฟ์„ ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค
79
+ # ๋ฌธ์„œ์— ๋”ฐ๋ฅด๋ฉด V7-Tekken ํ…œํ”Œ๋ฆฟ์ด ๊ธฐ๋ณธ ํ˜•์‹์ž…๋‹ˆ๋‹ค
80
+ return MessagesFormatterType.MISTRAL
81
  else:
82
  raise ValueError(f"Unsupported model: {model_name}")
83