yuzaa commited on
Commit
de9e1bb
1 Parent(s): f5965f6

Update modeling_minicpmv.py

Browse files
Files changed (1) hide show
  1. modeling_minicpmv.py +6 -0
modeling_minicpmv.py CHANGED
@@ -66,6 +66,12 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
66
  ]
67
  )
68
 
 
 
 
 
 
 
69
  def get_vision_embedding(self, pixel_values):
70
  res = []
71
  dtype = self.vpm.pos_embed.data.dtype
 
66
  ]
67
  )
68
 
69
+ def get_input_embeddings(self):
70
+ return self.llm.get_input_embeddings()
71
+
72
+ def set_input_embeddings(self, value):
73
+ self.llm.embed_tokens = value
74
+
75
  def get_vision_embedding(self, pixel_values):
76
  res = []
77
  dtype = self.vpm.pos_embed.data.dtype