Update vae/autoencoder_kl.py

vae/autoencoder_kl.py

@@ -19,6 +19,7 @@ from peft import LoraConfig
 
 from diffusers.configuration_utils import ConfigMixin, register_to_config
 # from diffusers.loaders import FromOriginalVAEMixin
+from diffusers.loaders import PeftAdapterMixin
 from diffusers.loaders.single_file_model import FromOriginalModelMixin
 from diffusers.utils.accelerate_utils import apply_forward_hook
 from diffusers.models.attention_processor import (
@@ -77,7 +78,7 @@ def my_vae_decoder_fwd(self, sample, latent_embeds=None):
     return sample
 
 
-class AutoencoderKL(ModelMixin, ConfigMixin, FromOriginalModelMixin):
+class AutoencoderKL(ModelMixin, ConfigMixin, FromOriginalModelMixin, PeftAdapterMixin):
     r"""
     A VAE model with KL loss for encoding images into latents and decoding latent representations into images.