Upload modeling_mic21.py with huggingface_hub

modeling_mic21.py

@@ -65,8 +65,8 @@ class MIC21SummarizerModel(PreTrainedModel):
         
     def forward(self, images, titles):
         pdb.set_trace()
-        prepared_images = self.components["image_processor"](images,return_tensors="pt", device="cuda")
-        prepared_images = prepared_images
+        prepared_images = self.components["image_processor"](images,return_tensors="pt")
+        prepared_images["pixel_values"] = prepared_images["pixel_values"].cuda()
         #prepared_images = prepared_images.to(f"cuda:{self.im_model_cuda_id}")
         
         img_features = self.components["image_model"](**prepared_images,output_hidden_states=True)