Spaces:
Running
on
Zero
Running
on
Zero
Update memo/models/motion_module.py
Browse files
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,
|