Update vae/autoencoder_kl.py

vae/autoencoder_kl.py

@@ -77,7 +77,7 @@ def my_vae_decoder_fwd(self, sample, latent_embeds=None):
     return sample
 
 
-class AutoencoderKL(ModelMixin, ConfigMixin, FromOriginalVAEMixin):
+class AutoencoderKL(ModelMixin, ConfigMixin, FromOriginalModelMixin):
     r"""
     A VAE model with KL loss for encoding images into latents and decoding latent representations into images.