mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-04 00:02:39 -08:00
Code Issues Projects Releases Packages Wiki Activity

initial refiner support

Browse source
This commit is contained in:
AUTOMATIC1111 2023-08-06 17:01:07 +03:00
parent 57e8a11d17
commit f1975b0213
6 changed files with 76 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

4
modules/processing.py
View file

@ -666,6 +666,10 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
stored_opts = {k: opts.data[k] for k in p.override_settings.keys()}
try:
# after running refiner, the refiner model is not unloaded - webui swaps back to main model here
if shared.sd_model.sd_checkpoint_info.title != opts.sd_model_checkpoint:
sd_models.reload_model_weights()
# if no checkpoint override or the override checkpoint can't be found, remove override entry and load opts checkpoint
if sd_models.checkpoint_aliases.get(p.override_settings.get('sd_model_checkpoint')) is None:
p.override_settings.pop('sd_model_checkpoint', None)