fix _always_upcast_modules for asym ae and vq_model

Author: a-r-r-o-w

src/diffusers/models/autoencoders/autoencoder_asym_kl.py

@@ -60,7 +60,7 @@ class AsymmetricAutoencoderKL(ModelMixin, ConfigMixin):
             Synthesis with Latent Diffusion Models](https://arxiv.org/abs/2112.10752) paper.
     """
 
-    _always_upcast_modules = ["MaskConditionDecoder"]
+    _always_upcast_modules = ["decoder"]
 
     @register_to_config
     def __init__(

src/diffusers/models/autoencoders/vq_model.py

@@ -71,7 +71,7 @@ class VQModel(ModelMixin, ConfigMixin):
             Type of normalization layer to use. Can be one of `"group"` or `"spatial"`.
     """
 
-    _always_upcast_modules = ["VectorQuantizer"]
+    _always_upcast_modules = ["quantize"]
 
     @register_to_config
     def __init__(