add DDIMScheduler in loadable classes

4569f758 · patil-suraj · a14d774b · 4569f758
Commit 4569f758 authored Jun 10, 2022 by patil-suraj
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/diffusers/pipeline_utils.py src/diffusers/pipeline_utils.py +1 -0

No files found.
--- a/src/diffusers/pipeline_utils.py
+++ b/src/diffusers/pipeline_utils.py
@@ -37,6 +37,7 @@ LOADABLE_CLASSES = {
        "GaussianDDPMScheduler": ["save_config", "from_config"],
        "ClassifierFreeGuidanceScheduler": ["save_config", "from_config"],
        "GlideDDIMScheduler": ["save_config", "from_config"],
+        "DDIMScheduler": ["save_config", "from_config"],
    },
    "transformers": {
        "PreTrainedTokenizer": ["save_pretrained", "from_pretrained"],