mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-07 11:00:43 -08:00
Code Issues Projects Releases Packages Wiki Activity

add refiner to StableDiffusionProcessing class

Browse source 
write out correct model name in infotext, rather than the refiner model
This commit is contained in:
AUTOMATIC1111 2023-08-13 06:07:30 +03:00
parent b2080756fc
commit fa9370b741
3 changed files with 36 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

16
modules/processing_scripts/refiner.py
View file

@ -41,15 +41,9 @@ class ScriptRefiner(scripts.Script):
def before_process(self, p, enable_refiner, refiner_checkpoint, refiner_switch_at):
# the actual implementation is in sd_samplers_common.py, apply_refiner
p.refiner_checkpoint_info = None
p.refiner_switch_at = None
if not enable_refiner or refiner_checkpoint in (None, "", "None"):
return
refiner_checkpoint_info = sd_models.get_closet_checkpoint_match(refiner_checkpoint)
if refiner_checkpoint_info is None:
raise Exception(f'Could not find checkpoint with name {refiner_checkpoint}')
p.refiner_checkpoint_info = refiner_checkpoint_info
p.refiner_switch_at = refiner_switch_at
p.refiner_checkpoint_info = None
p.refiner_switch_at = None
else:
p.refiner_checkpoint = refiner_checkpoint
p.refiner_switch_at = refiner_switch_at