pabloce commited on
Commit
9dce59b
·
verified ·
1 Parent(s): 6343090

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -91,6 +91,8 @@ def get_messages_formatter_type(model_name):
91
  return MessagesFormatterType.LLAMA_3
92
  elif "Mistral" in model_name:
93
  return MessagesFormatterType.MISTRAL
 
 
94
  else:
95
  raise ValueError(f"Unsupported model: {model_name}")
96
 
 
91
  return MessagesFormatterType.LLAMA_3
92
  elif "Mistral" in model_name:
93
  return MessagesFormatterType.MISTRAL
94
+ elif "Phi" in model_name:
95
+ return MessagesFormatterType.PHI_3
96
  else:
97
  raise ValueError(f"Unsupported model: {model_name}")
98