feature: update refiners

better handles img2img (partial diffusion runs)
brycedrennan · Jan 20, 2024 · 68795c0 · 68795c0
1 parent 4ae8427
commit 68795c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/imaginairy/api/generate_refiners.py b/imaginairy/api/generate_refiners.py
@@ -310,7 +310,7 @@ def latent_logger(latents):
                 noised_latent = init_latent
             else:
                 noised_latent = sd.scheduler.add_noise(
-                    x=init_latent, noise=noise, step=sd.steps[noise_step]
+                    x=init_latent, noise=noise, step=sd.scheduler.all_steps[noise_step]
                 )
 
         with lc.timing("text-conditioning"):