diff --git a/scripts/sd_dpmpp_sde_no_random.py b/scripts/sd_dpmpp_sde_no_random.py index 0a8e53f..633e222 100644 --- a/scripts/sd_dpmpp_sde_no_random.py +++ b/scripts/sd_dpmpp_sde_no_random.py @@ -24,7 +24,7 @@ def process(self, p): # Save the original "BrownianTreeNoiseSampler.__init__" p.orig_BrownianTreeNoiseSampler_init = k_diffusion.sampling.BrownianTreeNoiseSampler.__init__ # Create a hijacked version with the current seed as a literal and assign it to BrownianTreeNoiseSampler.__init__ - BrownianTreeNoiseSampler_init = create_hijacked_init(p.seed) + BrownianTreeNoiseSampler_init = create_hijacked_init(p.all_seeds[0]) k_diffusion.sampling.BrownianTreeNoiseSampler.__init__ = BrownianTreeNoiseSampler_init def postprocess(self, p, processed):