Change one-step dummy pipeline for testing (#1690)

Change the one-step dummy pipeline for testing
2022-12-13 16:55:49 +01:00 · 2022-12-13 16:55:49 +01:00 · e62dd5cfa8
parent 07f95503e5
commit e62dd5cfa8
1 changed files with 3 additions and 1 deletions
--- 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