diff --git a/modules/processing_vae.py b/modules/processing_vae.py index 60e8617dd..341e0e7e9 100644 --- a/modules/processing_vae.py +++ b/modules/processing_vae.py @@ -56,7 +56,7 @@ def full_vae_decode(latents, model): if upcast: if hasattr(model, 'upcast_vae'): # this is done by diffusers automatically if output_type != 'latent' model.upcast_vae() - model.vae = model.vae.to(dtype=torch.float32) + # model.vae = model.vae.to(dtype=torch.float32) latents = latents.to(torch.float32) else: latents = latents.to(devices.device)