Upload modeling_moment.py

modeling_moment.py

@@ -479,6 +479,8 @@ class MomentEmbeddingModel(MomentPreTrainedModel):
         )
         # [batch_size x n_channels x n_patches x d_model]
         hidden_states = hidden_states.reshape(batch_size, n_channels, n_patches, self.config.d_model)
+        print("＊input_mask_patch_view_for_hidden_states: ", input_mask_patch_view_for_hidden_states.shape)
+        print("hidden_states: ", hidden_states.shape)
         hidden_states = input_mask_patch_view_for_hidden_states * hidden_states
         # [batch_size, n_channels x n_patches, d_model]
         hidden_states = hidden_states.transpose(1, 2).reshape(batch_size, -1, self.config.d_model)