diff --git a/examples/community/one_step_unet.py b/examples/community/one_step_unet.py index 17107ca6eb..f3eaf1e0eb 100755 --- 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