fffiloni commited on
Commit
27b8649
·
verified ·
1 Parent(s): a3df9be

Update memo/models/motion_module.py

Browse files
Files changed (1) hide show
  1. memo/models/motion_module.py +4 -2
memo/models/motion_module.py CHANGED
@@ -368,10 +368,12 @@ class MemoryLinearAttention(Attention):
368
  else:
369
  raise NotImplementedError
370
 
 
 
371
  hidden_states = self.processor(
372
  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,
 
368
  else:
369
  raise NotImplementedError
370
 
371
+ assert "encoder_hidden_states" not in cross_attention_kwargs
372
+
373
  hidden_states = self.processor(
374
  self,
375
+ hidden_states,
376
+ motion_frames,
377
  encoder_hidden_states=encoder_hidden_states,
378
  attention_mask=attention_mask,
379
  n_motion_frames=n_motion_frames,