fixed a bug with seed reproducibility with bacth_size>0 and bacth_count>0
This commit is contained in:
parent
cfb6cc8ce5
commit
2430fc20f1
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue