qnguyen3 commited on
Commit
c11a322
1 Parent(s): c281bd9

Update modeling_llava_qwen2.py

Browse files
Files changed (1) hide show
  1. modeling_llava_qwen2.py +1 -1
modeling_llava_qwen2.py CHANGED
@@ -663,7 +663,7 @@ class LlavaMetaForCausalLM(ABC):
663
 
664
  def encode_images(self, images):
665
  image_features = self.get_model().get_vision_tower()(images)
666
- image_features = self.get_model().mm_projector(image_features)
667
  return image_features
668
 
669
  def prepare_inputs_labels_for_multimodal(
 
663
 
664
  def encode_images(self, images):
665
  image_features = self.get_model().get_vision_tower()(images)
666
+ image_features = self.get_model().mm_projector.to('cuda')(image_features)
667
  return image_features
668
 
669
  def prepare_inputs_labels_for_multimodal(