Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -108,7 +108,7 @@ def generate(prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition
|
|
108 |
|
109 |
if fim_mode:
|
110 |
output = prefix
|
111 |
-
elif
|
112 |
output = chat_prompt
|
113 |
else:
|
114 |
output = prompt
|
|
|
108 |
|
109 |
if fim_mode:
|
110 |
output = prefix
|
111 |
+
elif chat_mode:
|
112 |
output = chat_prompt
|
113 |
else:
|
114 |
output = prompt
|