Update memo/models/motion_module.py

memo/models/motion_module.py

@@ -368,12 +368,12 @@ class MemoryLinearAttention(Attention):
         else:
             raise NotImplementedError
 
-        assert "encoder_hidden_states" not in cross_attention_kwargs
         
+        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,