Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
c255fb2
1
Parent(s):
c5c3010
debug
Browse files
app.py
CHANGED
@@ -163,8 +163,8 @@ def http_bot(state):
|
|
163 |
|
164 |
# Construct prompt
|
165 |
cap_msgs, qa_msgs = build_messages(all_images, prompt)
|
166 |
-
cap_prompt = processor.apply_chat_template(
|
167 |
-
qa_prompt = processor.apply_chat_template(
|
168 |
image_tensor, _ = process_vision_info(cap_msgs)
|
169 |
cap_inputs = processor(text=cap_prompt, images=image_tensor, return_tensors="pt").to(mllm.device)
|
170 |
qa_inputs = processor(text=qa_prompt, images=image_tensor, return_tensors="pt").to(mllm.device)
|
|
|
163 |
|
164 |
# Construct prompt
|
165 |
cap_msgs, qa_msgs = build_messages(all_images, prompt)
|
166 |
+
cap_prompt = processor.apply_chat_template(cap_msgs, tokenize=False, add_generation_prompt=True)
|
167 |
+
qa_prompt = processor.apply_chat_template(qa_msgs, tokenize=False, add_generation_prompt=True)
|
168 |
image_tensor, _ = process_vision_info(cap_msgs)
|
169 |
cap_inputs = processor(text=cap_prompt, images=image_tensor, return_tensors="pt").to(mllm.device)
|
170 |
qa_inputs = processor(text=qa_prompt, images=image_tensor, return_tensors="pt").to(mllm.device)
|