aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_samplers_common.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-08-12 09:54:32 +0000
committerAUTOMATIC1111 <16777216c@gmail.com>2023-08-12 09:54:32 +0000
commitb293ed30610c040e621e1840d63047ae298f0650 (patch)
treeab85c6bea89827fe070f9adcd71c85355297d289 /modules/sd_samplers_common.py
parent64311faa6848d641cc452115e4e1eb47d2a7b519 (diff)
downloadstable-diffusion-webui-gfx803-b293ed30610c040e621e1840d63047ae298f0650.tar.gz
stable-diffusion-webui-gfx803-b293ed30610c040e621e1840d63047ae298f0650.tar.bz2
stable-diffusion-webui-gfx803-b293ed30610c040e621e1840d63047ae298f0650.zip
make it possible to use hires fix together with refiner
Diffstat (limited to 'modules/sd_samplers_common.py')
-rw-r--r--modules/sd_samplers_common.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py
index 85f3c7e0..40c7aae0 100644
--- a/modules/sd_samplers_common.py
+++ b/modules/sd_samplers_common.py
@@ -151,6 +151,9 @@ def apply_refiner(cfg_denoiser):
if refiner_checkpoint_info is None or shared.sd_model.sd_checkpoint_info == refiner_checkpoint_info:
return False
+ if getattr(cfg_denoiser.p, "enable_hr", False) and not cfg_denoiser.p.is_hr_pass:
+ return False
+
cfg_denoiser.p.extra_generation_params['Refiner'] = refiner_checkpoint_info.short_title
cfg_denoiser.p.extra_generation_params['Refiner switch at'] = refiner_switch_at