Spaces:
Running on Zero
Running on Zero
File size: 301 Bytes
855e1f5 | 1 2 3 4 5 6 7 8 9 10 11 | from diffusers import AutoencoderKL
from .._mixin import OstrisModelMixin
class KLVAE(AutoencoderKL, OstrisModelMixin):
"""The diffusers AutoencoderKL (SD/SDXL/Flux1/Z-Image image VAEs), loaded
from a checkpoint's vae/ subfolder through the universal loader."""
aitk_subfolder = "vae"
|