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

enhance metadata handling

Browse Source
This commit is contained in:
Vladimir Mandic
2024-05-15 10:08:31 -04:00
parent 69d7fef190
commit eb34241ca9
5 changed files with 35 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+1 -4
View File
@@ -676,10 +676,7 @@ def set_diffuser_options(sd_model, vae = None, op: str = 'model'):
sd_model.vae = vae
shared.log.debug(f'Setting {op} VAE: name={sd_vae.loaded_vae_file}')
if shared.opts.diffusers_vae_upcast != 'default':
if shared.opts.diffusers_vae_upcast == 'true':
sd_model.vae.config.force_upcast = True
else:
sd_model.vae.config.force_upcast = False
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)