support ltx2 gradient_checkpointing

2026-03-19 06:48:12 +00:00 · 2026-02-26 19:19:59 +08:00
parent a87910bc65
commit a18966c300
10 changed files with 36 additions and 33 deletions
--- a/diffsynth/utils/data/init.py
+++ b/diffsynth/utils/data/init.py
@@ -116,7 +116,7 @@ class VideoData:
        if self.height is not None and self.width is not None:
            return self.height, self.width
        else:
-            height, width, _ = self.__getitem__(0).shape
+            width, height = self.__getitem__(0).size
            return height, width

    def __getitem__(self, item):