correct library loading

7764669c · Patrick von Platen · 12b10cbe · 7764669c
Commit 7764669c authored Jun 12, 2022 by Patrick von Platen
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

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

No files found.
--- a/src/diffusers/pipeline_utils.py
+++ b/src/diffusers/pipeline_utils.py
@@ -34,11 +34,10 @@ logger = logging.get_logger(__name__)
 LOADABLE_CLASSES = {
    "diffusers": {
        "ModelMixin": ["save_pretrained", "from_pretrained"],
+        "SchedulerMixin": ["save_config", "from_config"],
        "DiffusionPipeline": ["save_pretrained", "from_pretrained"],
-        "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"],