fix a image cat bug in parallel vae (#66)

* fix a image cat bug in parallel vae

fix a image cat bug in parallel vae (#66)
* fix a image cat bug in parallel vae
c8374fec · Xinchi Huang · GitHub · 04eb738e · c8374fec
Commit c8374fec authored Jun 11, 2025 by Xinchi Huang Committed by GitHub Jun 11, 2025
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lightx2v/models/video_encoders/hf/wan/vae.py lightx2v/models/video_encoders/hf/wan/vae.py +1 -1

No files found.
--- a/lightx2v/models/video_encoders/hf/wan/vae.py
+++ b/lightx2v/models/video_encoders/hf/wan/vae.py
@@ -924,7 +924,7 @@ class WanVAE:

        torch.cuda.synchronize()

-        images = torch.cat(full_images, dim=-1)
+        images = torch.cat(full_images, dim=split_dim + 1)

        return images