Astaxanthin commited on
Commit
605f24b
·
verified ·
1 Parent(s): e608367

Upload modeling_keep.py

Browse files
Files changed (1) hide show
  1. modeling_keep.py +1 -1
modeling_keep.py CHANGED
@@ -66,7 +66,7 @@ class KEEPModel(PreTrainedModel):
66
  def forward(self, image_inputs, text_inputs):
67
  vision_features = self.encode_image(image_inputs)
68
 
69
- text_features = self.encode_image(text_inputs)
70
 
71
  # 返回两个独立的特征
72
  return {
 
66
  def forward(self, image_inputs, text_inputs):
67
  vision_features = self.encode_image(image_inputs)
68
 
69
+ text_features = self.encode_text(text_inputs)
70
 
71
  # 返回两个独立的特征
72
  return {