Update memo/models/motion_module.py

memo/models/motion_module.py

@@ -369,11 +369,11 @@ class MemoryLinearAttention(Attention):
             raise NotImplementedError
 
         
-        cross_attention_kwargs.pop("encoder_hidden_states", None)  # Extra safety
+        #cross_attention_kwargs.pop("encoder_hidden_states", None)  # Extra safety
         hidden_states = self.processor(
             self,
             hidden_states,
-            motion_frames=motion_frames,
+            motion_frames,
             encoder_hidden_states=encoder_hidden_states,
             attention_mask=attention_mask,
             n_motion_frames=n_motion_frames,