Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def chat(message, history, temperature, do_sample, max_tokens):
|
|
47 |
chat.append({"role": "assistant", "content": item[1]})
|
48 |
chat.append({"role": "user", "content": message})
|
49 |
messages = tok.apply_chat_template(chat, tokenize=False, add_generation_prompt=True)
|
50 |
-
model_inputs = tok([messages], return_tensors="pt").to(device)
|
51 |
streamer = TextIteratorStreamer(
|
52 |
tok, timeout=20.0, skip_prompt=True, skip_special_tokens=True
|
53 |
)
|
|
|
47 |
chat.append({"role": "assistant", "content": item[1]})
|
48 |
chat.append({"role": "user", "content": message})
|
49 |
messages = tok.apply_chat_template(chat, tokenize=False, add_generation_prompt=True)
|
50 |
+
model_inputs = tok([messages], return_tensors="pt",return_token_type_ids=False).to(device)
|
51 |
streamer = TextIteratorStreamer(
|
52 |
tok, timeout=20.0, skip_prompt=True, skip_special_tokens=True
|
53 |
)
|