Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -204,7 +204,7 @@ def http_bot(state, temperature, top_p, max_new_tokens, speaker):
|
|
| 204 |
# First round of conversation
|
| 205 |
if "llama3" in model_name.lower():
|
| 206 |
template_name = 'llama3_demo'
|
| 207 |
-
elif "
|
| 208 |
template_name = 'qwen2_demo'
|
| 209 |
else:
|
| 210 |
template_name = "default"
|
|
|
|
| 204 |
# First round of conversation
|
| 205 |
if "llama3" in model_name.lower():
|
| 206 |
template_name = 'llama3_demo'
|
| 207 |
+
elif "qwen-2" in model_name.lower():
|
| 208 |
template_name = 'qwen2_demo'
|
| 209 |
else:
|
| 210 |
template_name = "default"
|