openbmb
/

MiniCPM-Llama3-V-2_5

Image-Text-to-Text

feature-extraction

Model card Files Files and versions Community

qianyuchen commited on Jun 19

Commit

c5c2e9c

•

1 Parent(s): 45387f9

Update modeling_minicpmv.py

Files changed (1) hide show

modeling_minicpmv.py +2 -2

modeling_minicpmv.py CHANGED Viewed

@@ -69,8 +69,8 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
     def get_vllm_embedding(self, data):
         if 'vision_hidden_states' not in data:
-            dtype = self.vpm.embeddings.position_embedding.weight.dtype
-            device = self.vpm.embeddings.position_embedding.weight.device
             tgt_sizes = data['tgt_sizes']
             pixel_values_list = data['pixel_values']
             vision_hidden_states = []

     def get_vllm_embedding(self, data):
         if 'vision_hidden_states' not in data:
+            dtype = self.llm.lm_head.weight.dtype
+            device = self.llm.lm_head.weight.device
             tgt_sizes = data['tgt_sizes']
             pixel_values_list = data['pixel_values']
             vision_hidden_states = []