Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 08:44:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

cleanup lora detect

Browse Source
This commit is contained in:
Vladimir Mandic
2024-09-22 19:50:55 -04:00
parent f74187cdb1
commit ddefb38030
4 changed files with 57 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+1 -1
View File
@@ -59,7 +59,7 @@ def full_vae_decode(latents, model):
model.vae = model.vae.to(dtype=torch.float32)
latents = latents.to(torch.float32)
else:
latents = latents.to(model.vae.device)
latents = latents.to(devices.device)
if getattr(model.vae, "post_quant_conv", None) is not None:
latents = latents.to(next(iter(model.vae.post_quant_conv.parameters())).dtype)