diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-18 23:43:37 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-18 23:43:37 +0300 |
| commit | 2430fc20f10715b93a27638c21f83b2d3fa97398 (patch) | |
| tree | ccd0fc5ace6fd0e6b07cf87f7c6c4b63485a7bd8 /modules | |
| parent | cfb6cc8ce50b528ef9850ac671c3523121f70165 (diff) | |
fixed a bug with seed reproducibility with bacth_size>0 and bacth_count>0
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/sd_samplers.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 1b3dc302..af6811a1 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -255,6 +255,7 @@ class KDiffusionSampler: self.model_wrap_cfg.nmask = p.nmask
self.model_wrap_cfg.init_latent = p.init_latent
self.model_wrap.step = 0
+ self.sampler_noise_index = 0
if hasattr(k_diffusion.sampling, 'trange'):
k_diffusion.sampling.trange = lambda *args, **kwargs: extended_trange(*args, **kwargs)
@@ -269,6 +270,7 @@ class KDiffusionSampler: x = x * sigmas[0]
self.model_wrap_cfg.step = 0
+ self.sampler_noise_index = 0
if hasattr(k_diffusion.sampling, 'trange'):
k_diffusion.sampling.trange = lambda *args, **kwargs: extended_trange(*args, **kwargs)
|