Update modeling_minicpmv.py (#7)

- Update modeling_minicpmv.py (8d2b25a0833fa6e97000f2b987665e4a637e8066)


Co-authored-by: Sviatoslav Dovidchenko <[email protected]>

modeling_minicpmv.py

@@ -184,6 +184,8 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
                 vision_features = cur_vs_hs.view(-1, embed_dim)
 
                 updated_emb = cur_vllm_emb.clone()
+                vision_features = vision_features.to(cur_vllm_emb.device)
+
                 updated_emb.scatter_(0, indices_expanded, vision_features)
                 updated_vllm_embedding[i] = updated_emb
             elif self.training: