Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
|