Update app.py
Browse files
app.py
CHANGED
|
@@ -126,7 +126,7 @@ def talk(prompt, history):
|
|
| 126 |
# the chat template structure should be based on text generation model format
|
| 127 |
|
| 128 |
# indicates the end of a sequence
|
| 129 |
-
stream = model.create_chat_completion(messages = messages, stop=["</s>"], stream=False)
|
| 130 |
# print(f"{stream}")
|
| 131 |
print("check 7")
|
| 132 |
global historylog
|
|
|
|
| 126 |
# the chat template structure should be based on text generation model format
|
| 127 |
|
| 128 |
# indicates the end of a sequence
|
| 129 |
+
stream = model.create_chat_completion(messages = messages,max_tokens =400, stop=["</s>"], stream=False)
|
| 130 |
# print(f"{stream}")
|
| 131 |
print("check 7")
|
| 132 |
global historylog
|