From 1a37215a3df7cc26f5ad27718baea238d5160a96 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Fri, 27 Sep 2024 08:41:36 -0400 Subject: [PATCH] fix upcast --- modules/processing_vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)