Spaces:
Sleeping
Sleeping
aixsatoshi
commited on
Commit
•
184672e
1
Parent(s):
3a0dcd2
Update app.py
Browse files
app.py
CHANGED
@@ -54,7 +54,7 @@ def stream_chat(message: str, history: list, temperature: float, max_new_tokens:
|
|
54 |
conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
|
55 |
conversation.append({"role": "user", "content": message})
|
56 |
|
57 |
-
input_ids = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True).to(model.device)
|
58 |
#inputs = tokenizer(input_ids, return_tensors="pt").to(0)
|
59 |
|
60 |
streamer = TextIteratorStreamer(tokenizer, timeout=10., skip_prompt=True, skip_special_tokens=True)
|
|
|
54 |
conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
|
55 |
conversation.append({"role": "user", "content": message})
|
56 |
|
57 |
+
input_ids = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True,return_tensors="pt").to(model.device)
|
58 |
#inputs = tokenizer(input_ids, return_tensors="pt").to(0)
|
59 |
|
60 |
streamer = TextIteratorStreamer(tokenizer, timeout=10., skip_prompt=True, skip_special_tokens=True)
|