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