formatting

15e553b1 · Khalique Ahmed · ba7ba4df · 15e553b1
Commit 15e553b1 authored Nov 28, 2023 by Khalique Ahmed
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/diffusion/python_stable_diffusion_21/txt2img.py examples/diffusion/python_stable_diffusion_21/txt2img.py +1 -1

No files found.
--- a/examples/diffusion/python_stable_diffusion_21/txt2img.py
+++ b/examples/diffusion/python_stable_diffusion_21/txt2img.py
@@ -249,7 +249,7 @@ class StableDiffusionMGX():
                                                  noise_pred_uncond)

        # compute the previous noisy sample x_t -> x_t-1
-        return self.scheduler.step(torch.from_numpy(noise_pred), t,
+        return self.scheduler.step(torch.from_numpy(noise_pred), t, 
                                   latents).prev_sample

    @measure