fffiloni commited on
Commit
ca8e812
·
verified ·
1 Parent(s): 7ae37fd

Update memo/models/motion_module.py

Browse files
Files changed (1) hide show
  1. memo/models/motion_module.py +1 -1
memo/models/motion_module.py CHANGED
@@ -373,7 +373,7 @@ class MemoryLinearAttention(Attention):
373
  hidden_states = self.processor(
374
  self,
375
  hidden_states,
376
- # motion_frames=motion_frames,
377
  encoder_hidden_states=encoder_hidden_states,
378
  attention_mask=attention_mask,
379
  n_motion_frames=n_motion_frames,
 
373
  hidden_states = self.processor(
374
  self,
375
  hidden_states,
376
+ motion_frames=motion_frames,
377
  encoder_hidden_states=encoder_hidden_states,
378
  attention_mask=attention_mask,
379
  n_motion_frames=n_motion_frames,