Update modeling_llava_qwen2.py
Browse files- 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(
|