AutoencoderKLLTX2Video
The 3D variational autoencoder (VAE) model with KL loss used in LTX-2 was introduced by Lightricks.
The model can be loaded with the following code snippet.
from diffusers import AutoencoderKLLTX2Video
vae = AutoencoderKLLTX2Video.from_pretrained("Lightricks/LTX-2", subfolder="vae", torch_dtype=torch.float32).to("cuda")
AutoencoderKLLTX2Video
[[autodoc]] AutoencoderKLLTX2Video - decode - encode - all