feature: update refiners

better handles img2img (partial diffusion runs)
brycedrennan · Jan 20, 2024 · 6691967 · 6691967
1 parent 68795c0
commit 6691967
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/imaginairy/api/generate_refiners.py b/imaginairy/api/generate_refiners.py
@@ -306,7 +306,7 @@ def latent_logger(latents):
 
         if init_latent is not None:
             noise_step = noise_step if noise_step is not None else first_step
-            if first_step >= len(sd.steps):
+            if first_step >= len(sd.scheduler.all_steps):
                 noised_latent = init_latent
             else:
                 noised_latent = sd.scheduler.add_noise(

diff --git a/tests/expected_output/test_large_image_.png b/tests/expected_output/test_large_image_.png