eswardivi commited on
Commit
e08c569
·
verified ·
1 Parent(s): 90a750e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
  )