From e62dd5cfa8cf2d6c538366b0b9b85ce3fe613500 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 13 Dec 2022 16:55:49 +0100 Subject: [PATCH] Change one-step dummy pipeline for testing (#1690) Change the one-step dummy pipeline for testing --- examples/community/one_step_unet.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/community/one_step_unet.py b/examples/community/one_step_unet.py index 425d70fe23..17107ca6eb 100755 --- a/examples/community/one_step_unet.py +++ b/examples/community/one_step_unet.py @@ -18,5 +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 scheduler_output + return result