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

Update Stable Cascade pipeline

Browse Source
This commit is contained in:
Disty0
2024-02-28 19:41:42 +03:00
parent 537170fa2d
commit c8ed8659be
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+4 -2
View File
@@ -867,9 +867,11 @@ def load_diffuser(checkpoint_info=None, already_loaded_state_dict=None, timer=No
scheduler=decoder.scheduler,
vqgan=decoder.vqgan,
prior_prior=prior.prior,
prior_text_encoder=prior.text_encoder,
prior_tokenizer=prior.tokenizer,
prior_scheduler=prior.scheduler,
feature_extractor=prior.feature_extractor,
image_encoder=prior.image_encoder)
prior_feature_extractor=prior.feature_extractor,
prior_image_encoder=prior.image_encoder)
except Exception as e:
shared.log.error(f'Diffusers Failed loading {op}: {checkpoint_info.path} {e}')
if debug_load: