Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/pytorch3d.git" did not exist on "039e02601d447c127cc3a56c3d16716ef7d97a16"
Commit 4569f758 authored by patil-suraj's avatar patil-suraj
Browse files

add DDIMScheduler in loadable classes

parent a14d774b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/diffusers/pipeline_utils.py
View file @ 4569f758
...@@ -37,6 +37,7 @@ LOADABLE_CLASSES = { ...@@ -37,6 +37,7 @@ LOADABLE_CLASSES = {
"GaussianDDPMScheduler": ["save_config", "from_config"], "GaussianDDPMScheduler": ["save_config", "from_config"],
"ClassifierFreeGuidanceScheduler": ["save_config", "from_config"], "ClassifierFreeGuidanceScheduler": ["save_config", "from_config"],
"GlideDDIMScheduler": ["save_config", "from_config"], "GlideDDIMScheduler": ["save_config", "from_config"],
"DDIMScheduler": ["save_config", "from_config"],
}, },
"transformers": { "transformers": {
"PreTrainedTokenizer": ["save_pretrained", "from_pretrained"], "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