autoencoderkl_cosmos.md

AutoencoderKLCosmos

Cosmos Tokenizers.

Supported models:

• nvidia/Cosmos-1.0-Tokenizer-CV8x8x8

The model can be loaded with the following code snippet.

from diffusers import AutoencoderKLCosmos

vae = AutoencoderKLCosmos.from_pretrained("nvidia/Cosmos-1.0-Tokenizer-CV8x8x8", subfolder="vae")

AutoencoderKLCosmos

[[autodoc]] AutoencoderKLCosmos - decode - encode - all

AutoencoderKLOutput

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

DecoderOutput

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