aboutsummaryrefslogtreecommitdiff
path: root/modules/sd_samplers_kdiffusion.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-08-15 09:50:20 +0300
committerGitHub <noreply@github.com>2023-08-15 09:50:20 +0300
commitbc61ad9ec8cece0dc4ad445ee29abf4e9305f2df (patch)
treeb4aa6ca6232790900923c05370bdafeef5e41de0 /modules/sd_samplers_kdiffusion.py
parent79d4e81984171a047d6c71d97a67dda7dd87c43c (diff)
parent371b24b17c1cf98c9068a4b585b93cc1610702dc (diff)
Merge pull request #12564 from catboxanon/feat/img2img-noise
Add extra noise param for img2img operations
Diffstat (limited to 'modules/sd_samplers_kdiffusion.py')
-rw-r--r--modules/sd_samplers_kdiffusion.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py
index 08b9e740..08866e41 100644
--- a/modules/sd_samplers_kdiffusion.py
+++ b/modules/sd_samplers_kdiffusion.py
@@ -145,6 +145,10 @@ class KDiffusionSampler(sd_samplers_common.Sampler):
xi = x + noise * sigma_sched[0]
+ if opts.img2img_extra_noise > 0:
+ p.extra_generation_params["Extra noise"] = opts.img2img_extra_noise
+ xi += noise * opts.img2img_extra_noise
+
extra_params_kwargs = self.initialize(p)
parameters = inspect.signature(self.func).parameters