From c8ed8659be31176220a2b13d2371d3f89f0b85ac Mon Sep 17 00:00:00 2001 From: Disty0 Date: Wed, 28 Feb 2024 19:41:42 +0300 Subject: [PATCH] Update Stable Cascade pipeline --- modules/sd_models.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/sd_models.py b/modules/sd_models.py index 6e9e96fcf..f4795d537 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -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: