Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix upcsast

Browse Source
This commit is contained in:
Vladimir Mandic
2024-09-27 08:54:45 -04:00
parent 27afd84240
commit 199371314b
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+2 -1
View File
@@ -56,7 +56,8 @@ 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)
else:
model.vae = model.vae.to(dtype=torch.float32)
latents = latents.to(torch.float32)
else:
latents = latents.to(devices.device)