KaiChen1998 commited on
Commit
c255fb2
·
1 Parent(s): c5c3010
Files changed (1) hide show
  1. app.py +2 -2
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([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)
 
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)