fffiloni commited on
Commit
654d10a
·
verified ·
1 Parent(s): 044b2c5

Update memo/models/motion_module.py

Browse files
Files changed (1) hide show
  1. memo/models/motion_module.py +3 -3
memo/models/motion_module.py CHANGED
@@ -369,9 +369,9 @@ class MemoryLinearAttention(Attention):
369
  raise NotImplementedError
370
 
371
  hidden_states = self.processor(
372
- self,
373
- hidden_states,
374
- motion_frames,
375
  encoder_hidden_states=encoder_hidden_states,
376
  attention_mask=attention_mask,
377
  n_motion_frames=n_motion_frames,
 
369
  raise NotImplementedError
370
 
371
  hidden_states = self.processor(
372
+ attn=self,
373
+ hidden_states=hidden_states,
374
+ motion_frames=motion_frames,
375
  encoder_hidden_states=encoder_hidden_states,
376
  attention_mask=attention_mask,
377
  n_motion_frames=n_motion_frames,