rphrp1985 commited on
Commit
a724d1a
·
verified ·
1 Parent(s): 1513f9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -119,7 +119,7 @@ def respond(
119
  messages.append({"role": "user", "content": message})
120
  input_ids = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt").to(accelerator.device) #.to('cuda')
121
 
122
- input_ids2 = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True, return_tensors="pt").to(accelerator.device) #.to('cuda')
123
 
124
  input_str= str(input_ids2)
125
  print('input str = ', input_str)
 
119
  messages.append({"role": "user", "content": message})
120
  input_ids = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt").to(accelerator.device) #.to('cuda')
121
 
122
+ input_ids2 = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True, return_tensors="pt") #.to('cuda')
123
 
124
  input_str= str(input_ids2)
125
  print('input str = ', input_str)