Kiss3DGen / diffusers /docs /source /en /api /models /autoencoderkl_ltx_video.md
JiantaoLin
diffusers
df4a4de
|
raw
history blame
1.25 kB

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("TODO/TODO", 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