Skip to content

Commit c4f243d

Browse files
committed
fix: missing AutoencoderKL lora adapter
1 parent 0d1d267 commit c4f243d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/diffusers/models/autoencoders/autoencoder_kl.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -31,10 +31,11 @@
3131
)
3232
from ..modeling_outputs import AutoencoderKLOutput
3333
from ..modeling_utils import ModelMixin
34+
from ...loaders import PeftAdapterMixin
3435
from .vae import Decoder, DecoderOutput, DiagonalGaussianDistribution, Encoder
3536

3637

37-
class AutoencoderKL(ModelMixin, ConfigMixin, FromOriginalModelMixin):
38+
class AutoencoderKL(ModelMixin, ConfigMixin, FromOriginalModelMixin, PeftAdapterMixin):
3839
r"""
3940
A VAE model with KL loss for encoding images into latents and decoding latent representations into images.
4041

0 commit comments

Comments
 (0)