Skip to content

Commit b2ea8a8

Browse files
authored
Change PNDMPipeline to use PNDMScheduler (huggingface#2003)
* pndmpipeline uses pndmscheduler * formatted pipeline_pndm
1 parent 07c0fe4 commit b2ea8a8

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/diffusers/pipelines/pndm/pipeline_pndm.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,9 @@ class PNDMPipeline(DiffusionPipeline):
3939

4040
def __init__(self, unet: UNet2DModel, scheduler: PNDMScheduler):
4141
super().__init__()
42+
43+
scheduler = PNDMScheduler.from_config(scheduler.config)
44+
4245
self.register_modules(unet=unet, scheduler=scheduler)
4346

4447
@torch.no_grad()

0 commit comments

Comments
 (0)