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

Fix setting no-half-vae

Browse Source
This commit is contained in:
Disty0
2024-08-22 00:16:00 +03:00
parent 3c4d9f313d
commit 3a97db1f8a
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+4 -3
View File
@@ -678,10 +678,11 @@ def set_diffuser_options(sd_model, vae = None, op: str = 'model', offload=True):
shared.log.debug(f'Setting {op} VAE: name={sd_vae.loaded_vae_file}')
if shared.opts.diffusers_vae_upcast != 'default':
sd_model.vae.config.force_upcast = True if shared.opts.diffusers_vae_upcast == 'true' else False
if shared.opts.no_half_vae:
devices.dtype_vae = torch.float32
sd_model.vae.to(devices.dtype_vae)
shared.log.debug(f'Setting {op} VAE: upcast={sd_model.vae.config.force_upcast}')
if shared.opts.no_half_vae:
devices.dtype_vae = torch.float32
sd_model.vae.to(devices.dtype_vae)
shared.log.debug(f'Setting {op} VAE: no-half')
if hasattr(sd_model, "enable_vae_slicing"):
if shared.opts.diffusers_vae_slicing:
shared.log.debug(f'Setting {op}: enable VAE slicing')