ComfyUI/comfy/ldm/lightricks
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
..
vae ltx: vae: implement chunked encoder + CPU IO chunking (Big VRAM reductions) (#13062) 2026-03-19 09:58:47 -07:00
vocoders LTX audio vae novram fixes. (#12796) 2026-03-05 16:31:28 -05:00
av_model.py Support the LTXAV 2.3 model. (#12773) 2026-03-04 20:06:20 -05:00
embeddings_connector.py Support the LTXAV 2.3 model. (#12773) 2026-03-04 20:06:20 -05:00
latent_upsampler.py Support the LTXV 2 model. (#11632) 2026-01-05 01:58:59 -05:00
model.py Support the LTXAV 2.3 model. (#12773) 2026-03-04 20:06:20 -05:00
symmetric_patchifier.py Support the LTXV 2 model. (#11632) 2026-01-05 01:58:59 -05:00