fix a small typo in pipeline_ddpm.py (#948)

one small typo in pipeline_ddpm.py

just a small typo in one comment
This commit is contained in:
Chenguo Lin 2022-10-24 17:18:32 +08:00 committed by GitHub
parent 9bca40296e
commit 2d35f6733a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -79,7 +79,7 @@ class DDPMPipeline(DiffusionPipeline):
# 1. predict noise model_output
model_output = self.unet(image, t).sample
# 2. compute previous image: x_t -> t_t-1
# 2. compute previous image: x_t -> x_t-1
image = self.scheduler.step(model_output, t, image, generator=generator).prev_sample
image = (image / 2 + 0.5).clamp(0, 1)