From b2ea8a84e9db71c207d28687722d241447bde81e Mon Sep 17 00:00:00 2001
From: William Dalheim <willdalh@gmail.com>
Date: Mon, 16 Jan 2023 15:34:59 +0100
Subject: [PATCH] Change PNDMPipeline to use PNDMScheduler (#2003)

* pndmpipeline uses pndmscheduler

* formatted pipeline_pndm
---
 src/diffusers/pipelines/pndm/pipeline_pndm.py | 3 +++
 1 file changed, 3 insertions(+)

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()