Skip to content

S
Stable Diffusion Webui
Commit 2430fc20 authored by AUTOMATIC's avatar AUTOMATIC
Browse files
Options

fixed a bug with seed reproducibility with bacth_size>0 and bacth_count>0

parent cfb6cc8c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
modules/sd_samplers.py
View file @ 2430fc20
...@@ -255,6 +255,7 @@ class KDiffusionSampler: ...@@ -255,6 +255,7 @@ class KDiffusionSampler:
self.model_wrap_cfg.nmask = p.nmask self.model_wrap_cfg.nmask = p.nmask
self.model_wrap_cfg.init_latent = p.init_latent self.model_wrap_cfg.init_latent = p.init_latent
self.model_wrap.step = 0 self.model_wrap.step = 0
self.sampler_noise_index = 0
if hasattr(k_diffusion.sampling, 'trange'): if hasattr(k_diffusion.sampling, 'trange'):
k_diffusion.sampling.trange = lambda *args, **kwargs: extended_trange(*args, **kwargs) k_diffusion.sampling.trange = lambda *args, **kwargs: extended_trange(*args, **kwargs)
...@@ -269,6 +270,7 @@ class KDiffusionSampler: ...@@ -269,6 +270,7 @@ class KDiffusionSampler:
x = x * sigmas[0] x = x * sigmas[0]
self.model_wrap_cfg.step = 0 self.model_wrap_cfg.step = 0
self.sampler_noise_index = 0
if hasattr(k_diffusion.sampling, 'trange'): if hasattr(k_diffusion.sampling, 'trange'):
k_diffusion.sampling.trange = lambda *args, **kwargs: extended_trange(*args, **kwargs) k_diffusion.sampling.trange = lambda *args, **kwargs: extended_trange(*args, **kwargs)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment