alfredplpl
commited on
Commit
•
590e6c9
1
Parent(s):
e2f46ea
Update app.py
Browse files
app.py
CHANGED
@@ -69,7 +69,7 @@ def chat_llm_jp_v2(message: str,
|
|
69 |
conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
|
70 |
conversation.append({"role": "user", "content": message})
|
71 |
|
72 |
-
input_ids = tokenizer.apply_chat_template(conversation,
|
73 |
|
74 |
streamer = TextIteratorStreamer(tokenizer, timeout=10.0, skip_prompt=True, skip_special_tokens=True)
|
75 |
|
|
|
69 |
conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
|
70 |
conversation.append({"role": "user", "content": message})
|
71 |
|
72 |
+
input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt").to(model.device)
|
73 |
|
74 |
streamer = TextIteratorStreamer(tokenizer, timeout=10.0, skip_prompt=True, skip_special_tokens=True)
|
75 |
|