Update modeling_llava_qwen2.py

modeling_llava_qwen2.py

@@ -661,13 +661,11 @@ class LlavaMetaForCausalLM(ABC):
     def get_vision_tower(self):
         return self.get_model().get_vision_tower()
 
-    @spaces.GPU
     def encode_images(self, images):
-        image_features = self.get_model().get_vision_tower().to("cuda:0")(images)
-        image_features = self.get_model().mm_projector.to("cuda:0")(image_features)
+        image_features = self.get_model().get_vision_tower().to(decice="cuda:0", dtype=torch.float)(images)
+        image_features = self.get_model().mm_projector.to(device="cuda:0", dtype=torch.float)(image_features)
         return image_features
         
-    @spaces.GPU
     def prepare_inputs_labels_for_multimodal(
             self, input_ids, position_ids, attention_mask, past_key_values, labels, images
     ):