Spaces:
Sleeping
Sleeping
Commit
·
1288171
1
Parent(s):
7c2502a
debug
Browse files
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=
|
| 170 |
-
qa_inputs = processor(text=
|
| 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▌")
|