Update app.py
Browse files
app.py
CHANGED
@@ -151,7 +151,7 @@ def talk(prompt, history):
|
|
151 |
# print("check11")
|
152 |
# start a thread
|
153 |
outputs = []
|
154 |
-
input_ids =
|
155 |
|
156 |
start = time.time()
|
157 |
NUM_TOKENS=0
|
|
|
151 |
# print("check11")
|
152 |
# start a thread
|
153 |
outputs = []
|
154 |
+
input_ids = tokenizer(*messages)
|
155 |
|
156 |
start = time.time()
|
157 |
NUM_TOKENS=0
|