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

add models_not_to_quant option

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-09-11 12:31:07 -04:00
parent 2bb96e6926
commit 7940217764
15 changed files with 59 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+4
View File
@@ -137,6 +137,10 @@ def full_vae_decode(latents, model):
latents = latents.to(devices.dtype_vae)
else:
latents = latents.to(next(iter(model.vae.post_quant_conv.parameters())).dtype)
# if getattr(model.vae.post_quant_conv, "bias", None) is not None:
# model.vae.post_quant_conv.bias = torch.nn.Parameter(model.vae.post_quant_conv.bias.to(devices.device), requires_grad=False)
# if getattr(model.vae.post_quant_conv, "weight", None) is not None:
# model.vae.post_quant_conv.weight = torch.nn.Parameter(model.vae.post_quant_conv.weight.to(devices.device), requires_grad=False)
else:
latents = latents.to(model.vae.dtype)