Skip to content

Commit f6f1ec3

Browse files
allow loading ddpm models into ddim (huggingface#1932)
1 parent beb932c commit f6f1ec3

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/diffusers/pipelines/ddim/pipeline_ddim.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616

1717
import torch
1818

19+
from ...schedulers import DDIMScheduler
1920
from ...utils import deprecate, randn_tensor
2021
from ..pipeline_utils import DiffusionPipeline, ImagePipelineOutput
2122

@@ -34,6 +35,10 @@ class DDIMPipeline(DiffusionPipeline):
3435

3536
def __init__(self, unet, scheduler):
3637
super().__init__()
38+
39+
# make sure scheduler can always be converted to DDIM
40+
scheduler = DDIMScheduler.from_config(scheduler.config)
41+
3742
self.register_modules(unet=unet, scheduler=scheduler)
3843

3944
@torch.no_grad()

0 commit comments

Comments
 (0)