AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Change PNDMPipeline to use PNDMScheduler (#2003)

Browse Source 
* pndmpipeline uses pndmscheduler

* formatted pipeline_pndm
This commit is contained in:
William Dalheim
2023-01-16 15:34:59 +01:00
committed by GitHub
parent 07c0fe4b87
commit b2ea8a84e9
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/diffusers/pipelines/pndm/pipeline_pndm.py
View File

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