make style

2022-12-13 17:01:18 +00:00 · 2022-12-13 17:01:18 +00:00 · 8b7cb962a5
parent e1bb8f6188
commit 8b7cb962a5
1 changed files with 1 additions and 1 deletions
--- 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