fix(api): use correct latents for scheduler noise

ssube · Apr 27, 2023 · bc946bb · bc946bb
1 parent bad00dc
commit bc946bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/diffusers/pipelines/panorama.py b/api/onnx_web/diffusers/pipelines/panorama.py
@@ -651,7 +651,7 @@ def img2img(
 
         noise = generator.randn(*latents.shape).astype(latents_dtype)
         latents = self.scheduler.add_noise(
-            torch.from_numpy(init_latents), torch.from_numpy(noise), torch.from_numpy(timesteps)
+            torch.from_numpy(latents), torch.from_numpy(noise), torch.from_numpy(timesteps)
         )
         latents = latents.numpy()