Skip to content

Commit e62dd5c

Browse files
Change one-step dummy pipeline for testing (huggingface#1690)
Change the one-step dummy pipeline for testing
1 parent 07f9550 commit e62dd5c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

examples/community/one_step_unet.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,5 +18,7 @@ def __call__(self):
1818

1919
model_output = self.unet(image, timestep).sample
2020
scheduler_output = self.scheduler.step(model_output, timestep, image).prev_sample
21+
22+
result = scheduler_output - scheduler_output + torch.ones_like(scheduler_output)
2123

22-
return scheduler_output
24+
return result

0 commit comments

Comments
 (0)