Upload modeling_mic21.py with huggingface_hub

modeling_mic21.py

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