Update app.py
Browse files
app.py
CHANGED
|
@@ -82,7 +82,7 @@ def generate(
|
|
| 82 |
if any([end_token in response.token.text for end_token in [EOS_STRING, EOT_STRING]]):
|
| 83 |
if fim_mode:
|
| 84 |
output += suffix
|
| 85 |
-
|
| 86 |
else:
|
| 87 |
return output
|
| 88 |
else:
|
|
|
|
| 82 |
if any([end_token in response.token.text for end_token in [EOS_STRING, EOT_STRING]]):
|
| 83 |
if fim_mode:
|
| 84 |
output += suffix
|
| 85 |
+
print("output", output)
|
| 86 |
else:
|
| 87 |
return output
|
| 88 |
else:
|