Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/fairseq.git" did not exist on "ccb5dea58dc556549df00453a144d382ffb727b6"
Commit 8b7cb962 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent e1bb8f61
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/community/one_step_unet.py
View file @ 8b7cb962
...@@ -18,7 +18,7 @@ class UnetSchedulerOneForwardPipeline(DiffusionPipeline): ...@@ -18,7 +18,7 @@ class UnetSchedulerOneForwardPipeline(DiffusionPipeline):
model_output = self.unet(image, timestep).sample model_output = self.unet(image, timestep).sample
scheduler_output = self.scheduler.step(model_output, timestep, image).prev_sample scheduler_output = self.scheduler.step(model_output, timestep, image).prev_sample
result = scheduler_output - scheduler_output + torch.ones_like(scheduler_output) result = scheduler_output - scheduler_output + torch.ones_like(scheduler_output)
return result 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