Unverified Commit e62dd5cf authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

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

Change the one-step dummy pipeline for testing
parent 07f95503
......@@ -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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment