make style

8b7cb962 · Patrick von Platen · e1bb8f61 · 8b7cb962
Commit 8b7cb962 authored Dec 13, 2022 by Patrick von Platen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/community/one_step_unet.py examples/community/one_step_unet.py +1 -1

No files found.
--- a/examples/community/one_step_unet.py
+++ b/examples/community/one_step_unet.py
@@ -18,7 +18,7 @@ class UnetSchedulerOneForwardPipeline(DiffusionPipeline):

        model_output = self.unet(image, timestep).sample
        scheduler_output = self.scheduler.step(model_output, timestep, image).prev_sample
-        
+
        result = scheduler_output - scheduler_output + torch.ones_like(scheduler_output)

        return result