Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "f442955c6e871dcaf7d4003f74970e0905c2ff27"
Commit 606ac57e authored by Patrick von Platen's avatar Patrick von Platen
Browse files

finish pndm sampler

parent 394243ce
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 10 deletions
src/diffusers/pipelines/pndm/pipeline_pndm.py
View file @ 606ac57e
...@@ -49,16 +49,6 @@ class PNDMPipeline(DiffusionPipeline): ...@@ -49,16 +49,6 @@ class PNDMPipeline(DiffusionPipeline):
image = self.scheduler.step(model_output, t, image)["prev_sample"] image = self.scheduler.step(model_output, t, image)["prev_sample"]
# for i, t in enumerate(tqdm(self.scheduler.prk_timesteps)):
# model_output = self.unet(image, t)["sample"]
#
# image = self.scheduler.step_prk(model_output, t, image, i=i)["prev_sample"]
#
# for i, t in enumerate(tqdm(self.scheduler.plms_timesteps)):
# model_output = self.unet(image, t)["sample"]
#
# image = self.scheduler.step_plms(model_output, t, image, i=i)["prev_sample"]
image = (image / 2 + 0.5).clamp(0, 1) image = (image / 2 + 0.5).clamp(0, 1)
image = image.cpu().permute(0, 2, 3, 1).numpy() image = image.cpu().permute(0, 2, 3, 1).numpy()
if output_type == "pil": if output_type == "pil":
......
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