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

remove non-blocking

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-12-06 09:15:37 -05:00
parent 2a1fbd904f
commit 28eeb00432
6 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+1 -1
View File
@@ -117,7 +117,7 @@ def full_vae_decode(latents, model):
model.vae.orig_dtype = model.vae.dtype
model.vae = model.vae.to(dtype=torch.float32)
latents = latents.to(torch.float32)
latents = latents.to(devices.device, non_blocking=True)
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)