Spaces:
Running
on
Zero
Running
on
Zero
Update modeling_llava_qwen2.py
Browse files- modeling_llava_qwen2.py +2 -2
modeling_llava_qwen2.py
CHANGED
@@ -662,8 +662,8 @@ class LlavaMetaForCausalLM(ABC):
|
|
662 |
return self.get_model().get_vision_tower()
|
663 |
|
664 |
def encode_images(self, images):
|
665 |
-
image_features = self.get_model().get_vision_tower()
|
666 |
-
image_features = self.get_model().mm_projector
|
667 |
return image_features
|
668 |
|
669 |
def prepare_inputs_labels_for_multimodal(
|
|
|
662 |
return self.get_model().get_vision_tower()
|
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(
|