Update modeling_llava_qwen2.py

modeling_llava_qwen2.py

@@ -663,8 +663,8 @@ class LlavaMetaForCausalLM(ABC):
 
     @spaces.GPU
     def encode_images(self, images):
-        image_features = self.get_model().get_vision_tower()(images)
-        image_features = self.get_model().mm_projector(image_features)
+        image_features = self.get_model().get_vision_tower().to("cuda:0")(images)
+        image_features = self.get_model().mm_projector.to("cuda:0")(image_features)
         return image_features
         
     @spaces.GPU