ButterChicken98's picture
End of training
d4c007b verified

AutoencoderKLLTXVideo

The 3D variational autoencoder (VAE) model with KL loss used in LTX was introduced by Lightricks.

The model can be loaded with the following code snippet.

from diffusers import AutoencoderKLLTXVideo

vae = AutoencoderKLLTXVideo.from_pretrained("Lightricks/LTX-Video", subfolder="vae", torch_dtype=torch.float32).to("cuda")

AutoencoderKLLTXVideo

[[autodoc]] AutoencoderKLLTXVideo - decode - encode - all

AutoencoderKLOutput

[[autodoc]] models.autoencoders.autoencoder_kl.AutoencoderKLOutput

DecoderOutput

[[autodoc]] models.autoencoders.vae.DecoderOutput