11 lines
313 B
Python
11 lines
313 B
Python
from diffusers import AutoencoderKLFlux2
|
|
|
|
from .._mixin import OstrisModelMixin
|
|
|
|
|
|
class Flux2KLVAE(AutoencoderKLFlux2, OstrisModelMixin):
|
|
"""The diffusers Flux2 KL VAE (ernie_image; distinct from the hand-rolled
|
|
BFL AutoEncoder in vae/flux2_kl.py that flux2/ideogram4 use)."""
|
|
|
|
aitk_subfolder = "vae"
|