Spaces:

HTBris
/

zerogpu-sdxlTrainer

Running on Zero

zerogpu-sdxlTrainer / toolkit /models /v2 /vae /autoencoder_kl.py

Deploy Ostris AI-Toolkit ZeroGPU Trainer (Illustrious XL / SDXL / FLUX.1)

855e1f5 3 days ago

301 Bytes

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"

	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"