Update app.py
Browse files
app.py
CHANGED
|
@@ -154,8 +154,9 @@ def talk(prompt, history):
|
|
| 154 |
outputs = []
|
| 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
|
| 161 |
print('-'*4+'Start Generation'+'-'*4)
|
|
|
|
| 154 |
outputs = []
|
| 155 |
print(messages)
|
| 156 |
print(*messages)
|
| 157 |
+
print(model.tokenize(messages))
|
| 158 |
# input_ids = tokenizer(*messages)
|
| 159 |
+
# print(model.generate(tensor([[ 1, 529, 29989, 5205, 29989]])))
|
| 160 |
start = time.time()
|
| 161 |
NUM_TOKENS=0
|
| 162 |
print('-'*4+'Start Generation'+'-'*4)
|