We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e1bb8f6 commit 8b7cb96Copy full SHA for 8b7cb96
examples/community/one_step_unet.py
@@ -18,7 +18,7 @@ def __call__(self):
18
19
model_output = self.unet(image, timestep).sample
20
scheduler_output = self.scheduler.step(model_output, timestep, image).prev_sample
21
-
+
22
result = scheduler_output - scheduler_output + torch.ones_like(scheduler_output)
23
24
return result
0 commit comments