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

enable sdxl vae

Browse Source
This commit is contained in:
Vladimir Mandic
2023-07-16 17:58:08 -04:00
parent a9e2d90a47
commit 4e48173e37
8 changed files with 74 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+5 -4
View File
@@ -638,10 +638,11 @@ def load_diffuser(checkpoint_info=None, already_loaded_state_dict=None, timer=No
return
shared.log.info(f'Loading diffuser {op}: {checkpoint_info.filename}')
vae_file, vae_source = sd_vae.resolve_vae(checkpoint_info.filename)
vae = sd_vae.load_vae_diffusers(None, vae_file, vae_source)
if vae is not None:
diffusers_load_config["vae"] = vae
if op == 'model':
vae_file, vae_source = sd_vae.resolve_vae(checkpoint_info.filename)
vae = sd_vae.load_vae_diffusers(None, vae_file, vae_source)
if vae is not None:
diffusers_load_config["vae"] = vae
if not os.path.isfile(checkpoint_info.path):
try: