Unverified Commit b2ea8a84 authored by William Dalheim's avatar William Dalheim Committed by GitHub
Browse files

Change PNDMPipeline to use PNDMScheduler (#2003)

* pndmpipeline uses pndmscheduler

* formatted pipeline_pndm
parent 07c0fe4b
...@@ -39,6 +39,9 @@ class PNDMPipeline(DiffusionPipeline): ...@@ -39,6 +39,9 @@ class PNDMPipeline(DiffusionPipeline):
def __init__(self, unet: UNet2DModel, scheduler: PNDMScheduler): def __init__(self, unet: UNet2DModel, scheduler: PNDMScheduler):
super().__init__() super().__init__()
scheduler = PNDMScheduler.from_config(scheduler.config)
self.register_modules(unet=unet, scheduler=scheduler) self.register_modules(unet=unet, scheduler=scheduler)
@torch.no_grad() @torch.no_grad()
......
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