Update app.py
Browse files
app.py
CHANGED
|
@@ -155,7 +155,6 @@ def talk(prompt, history):
|
|
| 155 |
print(messages)
|
| 156 |
print(*messages)
|
| 157 |
# input_ids = tokenizer(*messages)
|
| 158 |
-
|
| 159 |
print(model.generate(tensor([[ 1, 529, 29989, 5205, 29989]])))
|
| 160 |
start = time.time()
|
| 161 |
NUM_TOKENS=0
|
|
|
|
| 155 |
print(messages)
|
| 156 |
print(*messages)
|
| 157 |
# input_ids = tokenizer(*messages)
|
|
|
|
| 158 |
print(model.generate(tensor([[ 1, 529, 29989, 5205, 29989]])))
|
| 159 |
start = time.time()
|
| 160 |
NUM_TOKENS=0
|