andrewqian123
/

LLAMA_BATCH

Visual Question Answering

Model card Files Files and versions

andrewqian123 commited on Sep 3, 2024

Commit

7531a10

·

verified ·

1 Parent(s): 6ef8720

Update modeling_minicpmv.py

Files changed (1) hide show

modeling_minicpmv.py +1 -1

modeling_minicpmv.py CHANGED Viewed

@@ -266,7 +266,7 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
             print(model_inputs)
             (
                 input_embeds,
-                vision_hidden_states,
             ) = self.get_vllm_embedding(model_inputs)
             print(input_embeds.shape, f"INPUT_EMBEDS {counter}")
             counter += 1

             print(model_inputs)
             (
                 input_embeds,
+                vision_hidden_states_dummy,
             ) = self.get_vllm_embedding(model_inputs)
             print(input_embeds.shape, f"INPUT_EMBEDS {counter}")
             counter += 1