KaiChen1998 commited on
Commit
1288171
·
1 Parent(s): 7c2502a
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -166,8 +166,8 @@ def http_bot(state):
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)
171
 
172
  # Step 1: Tentative Response
173
  state.append_message(state.roles[1], "# Tentative Response\n\n▌")
 
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)
171
 
172
  # Step 1: Tentative Response
173
  state.append_message(state.roles[1], "# Tentative Response\n\n▌")