Upload modeling_moment.py

modeling_moment.py

@@ -448,7 +448,7 @@ class MomentEmbeddingModel(MomentPreTrainedModel):
         attention_mask = patch_view_mask.repeat_interleave(n_channels, dim=0)
         outputs = self.encoder(inputs_embeds=enc_in, attention_mask=attention_mask)
         enc_out = outputs.last_hidden_state
-        hidden_states = outputs.hidden_states  # hidden_statesを取得
+        hidden_states = outputs.last_hidden_state  # hidden_statesを取得
 
         enc_out = enc_out.reshape((-1, n_channels, n_patches, self.config.d_model))
         # [batch_size x n_channels x n_patches x d_model]