ComfyUI/comfy/ldm/lightricks/vae
rattus fabed694a2
ltx: vae: implement chunked encoder + CPU IO chunking (Big VRAM reductions) (#13062)
* ltx: vae: add cache state to downsample block

* ltx: vae: Add time stride awareness to causal_conv_3d

* ltx: vae: Automate truncation for encoder

Other VAEs just truncate without error. Do the same.

* sd/ltx: Make chunked_io a flag in its own right

Taking this bi-direcitonal, so make it a for-purpose named flag.

* ltx: vae: implement chunked encoder + CPU IO chunking

People are doing things with big frame counts in LTX including V2V
flows. Implement the time-chunked encoder to keep the VRAM down, with
the converse of the new CPU pre-allocation technique, where the chunks
are brought from the CPU JIT.

* ltx: vae-encode: round chunk sizes more strictly

Only powers of 2 and multiple of 8 are valid due to cache slicing.
2026-03-19 09:58:47 -07:00
..
audio_vae.py Support the LTXAV 2.3 model. (#12773) 2026-03-04 20:06:20 -05:00
causal_audio_autoencoder.py Support the LTXAV 2.3 model. (#12773) 2026-03-04 20:06:20 -05:00
causal_conv3d.py ltx: vae: implement chunked encoder + CPU IO chunking (Big VRAM reductions) (#13062) 2026-03-19 09:58:47 -07:00
causal_video_autoencoder.py ltx: vae: implement chunked encoder + CPU IO chunking (Big VRAM reductions) (#13062) 2026-03-19 09:58:47 -07:00
conv_nd_factory.py Support LTXV 0.9.5. 2025-03-05 00:13:49 -05:00
dual_conv3d.py Support LTXV 0.9.5. 2025-03-05 00:13:49 -05:00
pixel_norm.py Support Lightricks LTX-Video model. 2024-11-22 08:46:39 -05:00