diff options
| author | Kyle <zerouex@gmail.com> | 2023-02-03 19:15:32 -0500 |
|---|---|---|
| committer | Kyle <zerouex@gmail.com> | 2023-02-03 19:15:32 -0500 |
| commit | c27c0de0f73c5f533acfa10426dbac7ac988bc85 (patch) | |
| tree | b024b9959bed07d1a093b229af7b0b9e166b1020 /modules | |
| parent | 6c6c6636bb123d664999c888cda47a1f8bad635b (diff) | |
txt2img Hires Fix
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/processing.py | 1 | ||||
| -rw-r--r-- | modules/sd_samplers_kdiffusion.py | 2 |
2 files changed, 2 insertions, 1 deletions
diff --git a/modules/processing.py b/modules/processing.py index c33694cc..e1b53ac0 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -268,6 +268,7 @@ class Processed: self.height = p.height
self.sampler_name = p.sampler_name
self.cfg_scale = p.cfg_scale
+ self.image_cfg_scale = getattr(p, 'image_cfg_scale', None)
self.steps = p.steps
self.batch_size = p.batch_size
self.restore_faces = p.restore_faces
diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py index a16ba69b..6107e99e 100644 --- a/modules/sd_samplers_kdiffusion.py +++ b/modules/sd_samplers_kdiffusion.py @@ -352,7 +352,7 @@ class KDiffusionSampler: 'cond_scale': p.cfg_scale,
}
- if p.image_cfg_scale:
+ if hasattr(p, 'image_cfg_scale'):
extra_args['image_cfg_scale'] = p.image_cfg_scale
samples = self.launch_sampling(t_enc + 1, lambda: self.func(self.model_wrap_cfg, xi, extra_args=extra_args, disable=False, callback=self.callback_state, **extra_params_kwargs))
|