diff --git a/src/diffusers/pipelines/pndm/pipeline_pndm.py b/src/diffusers/pipelines/pndm/pipeline_pndm.py index 39cb705123..d39995097a 100644 --- a/src/diffusers/pipelines/pndm/pipeline_pndm.py +++ b/src/diffusers/pipelines/pndm/pipeline_pndm.py @@ -39,6 +39,9 @@ class PNDMPipeline(DiffusionPipeline): def __init__(self, unet: UNet2DModel, scheduler: PNDMScheduler): super().__init__() + + scheduler = PNDMScheduler.from_config(scheduler.config) + self.register_modules(unet=unet, scheduler=scheduler) @torch.no_grad()