Skip to content

Commit 8b7cb96

Browse files
make style
1 parent e1bb8f6 commit 8b7cb96

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/community/one_step_unet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +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-
21+
2222
result = scheduler_output - scheduler_output + torch.ones_like(scheduler_output)
2323

2424
return result

0 commit comments

Comments
 (0)