mirror of
https://github.com/THUDM/CogVideo.git
synced 2025-06-18 14:39:17 +08:00
Update dit_video_concat.py
This commit is contained in:
parent
cd590183f9
commit
db498073d0
@ -309,7 +309,6 @@ class Rotary3DPositionEmbeddingMixin(BaseMixin):
|
||||
|
||||
def position_embedding_forward(self, position_ids, **kwargs):
|
||||
if self.pos_embedding is not None:
|
||||
print(self.pos_embedding)
|
||||
return self.pos_embedding[:, :self.text_length + kwargs["seq_length"]]
|
||||
else:
|
||||
return None
|
||||
|
Loading…
x
Reference in New Issue
Block a user