back to vanilla

memo/models/motion_module.py

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