Update cumo/model/multimodal_encoder/clip_encoder.py

cumo/model/multimodal_encoder/clip_encoder.py

@@ -86,7 +86,7 @@ class CLIPVisionTower(nn.Module):
             for image in images:
                 dev = image.device if hasattr(image, "device") else torch.device("cuda" if torch.cuda.is_available() else "cpu")
                 dt = image.dtype if hasattr(image, "dtype") else torch.float16
-                image_forward_out = self.vision_model(image.to(device=dev, dtype=dt).unsqueeze(0), output_hidden_states=True)
+                image_forward_out = self.vision_model(image.to(device=dev, dtype=dt).unsqueeze(0))
                 image_feature = self.feature_select(image_forward_out).to(image.dtype)
                 image_features.append(image_feature)