Update app.py
Browse files
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")
|
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)
|