diff --git a/modules/processing_diffusers.py b/modules/processing_diffusers.py index a00528e65..3b9a207f0 100644 --- a/modules/processing_diffusers.py +++ b/modules/processing_diffusers.py @@ -243,13 +243,6 @@ def process_diffusers(p: processing.StableDiffusionProcessing): args["prior_guidance_scale"] = p.cfg_scale if 'decoder_guidance_scale' in possible: args["decoder_guidance_scale"] = p.image_cfg_scale - # TODO Stable Cascade callbacks are currently broken in combined pipeline so preview will not get triggered - if 'prior_callback_on_step_end' in possible: - possible.remove('callback_on_step_end') - if 'callback_on_step_end' in possible: - possible.remove('callback_on_step_end') - if 'callback' in possible: - possible.remove('callback') # set callbacks if 'callback_steps' in possible: diff --git a/modules/sd_models.py b/modules/sd_models.py index 3911b6056..0291b9adb 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -918,7 +918,7 @@ def load_diffuser(checkpoint_info=None, already_loaded_state_dict=None, timer=No diffusers_load_config.pop("variant", None) decoder = diffusers.StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", cache_dir=shared.opts.diffusers_dir, revision="refs/pr/44", **diffusers_load_config) shared.log.debug(f'StableCascade decoder: scale={decoder.latent_dim_scale}') - prior = diffusers.StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", cache_dir=shared.opts.diffusers_dir, **diffusers_load_config) + prior = diffusers.StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", cache_dir=shared.opts.diffusers_dir, revision="refs/pr/2", **diffusers_load_config) shared.log.debug(f'StableCascade prior: scale={prior.resolution_multiple}') sd_model = diffusers.StableCascadeCombinedPipeline( tokenizer=decoder.tokenizer, diff --git a/wiki b/wiki index 657c48a5c..365b15ca1 160000 --- a/wiki +++ b/wiki @@ -1 +1 @@ -Subproject commit 657c48a5cfab058fd046c682c0cb70ece3789073 +Subproject commit 365b15ca100eb9abafb1f702e2a9abcb803618c4