Update dit_video_concat.py

This commit is contained in:
zR 2024-09-22 14:51:51 +08:00
parent cd590183f9
commit db498073d0

View File

@ -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