We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent beb932c commit f6f1ec3Copy full SHA for f6f1ec3
src/diffusers/pipelines/ddim/pipeline_ddim.py
@@ -16,6 +16,7 @@
16
17
import torch
18
19
+from ...schedulers import DDIMScheduler
20
from ...utils import deprecate, randn_tensor
21
from ..pipeline_utils import DiffusionPipeline, ImagePipelineOutput
22
@@ -34,6 +35,10 @@ class DDIMPipeline(DiffusionPipeline):
34
35
36
def __init__(self, unet, scheduler):
37
super().__init__()
38
+
39
+ # make sure scheduler can always be converted to DDIM
40
+ scheduler = DDIMScheduler.from_config(scheduler.config)
41
42
self.register_modules(unet=unet, scheduler=scheduler)
43
44
@torch.no_grad()
0 commit comments