Commit 4569f758 authored by patil-suraj's avatar patil-suraj
Browse files

add DDIMScheduler in loadable classes

parent a14d774b
......@@ -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"],
......
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