11 lines
301 B
Python
11 lines
301 B
Python
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"
|