yuzaa commited on
Commit
f92aff2
1 Parent(s): e4f0bfd

Update modeling_minicpmv.py

Browse files
Files changed (1) hide show
  1. modeling_minicpmv.py +6 -0
modeling_minicpmv.py CHANGED
@@ -61,6 +61,12 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
61
  ]
62
  )
63
 
 
 
 
 
 
 
64
  def get_vllm_embedding(self, data):
65
  if 'vision_hidden_states' not in data:
66
  dtype = self.vpm.embeddings.position_embedding.weight.dtype
 
61
  ]
62
  )
63
 
64
+ def get_input_embeddings(self):
65
+ return self.llm.get_input_embeddings()
66
+
67
+ def set_input_embeddings(self, value):
68
+ self.llm.embed_tokens = value
69
+
70
  def get_vllm_embedding(self, data):
71
  if 'vision_hidden_states' not in data:
72
  dtype = self.vpm.embeddings.position_embedding.weight.dtype