Skip to content

GitLab

Commit 7764669c authored by Patrick von Platen's avatar Patrick von Platen
Browse files

correct library loading

parent 12b10cbe
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
src/diffusers/pipeline_utils.py
View file @ 7764669c
...@@ -34,11 +34,10 @@ logger = logging.get_logger(__name__) ...@@ -34,11 +34,10 @@ logger = logging.get_logger(__name__)
LOADABLE_CLASSES = { LOADABLE_CLASSES = {
"diffusers": { "diffusers": {
"ModelMixin": ["save_pretrained", "from_pretrained"], "ModelMixin": ["save_pretrained", "from_pretrained"],
"SchedulerMixin": ["save_config", "from_config"],
"DiffusionPipeline": ["save_pretrained", "from_pretrained"], "DiffusionPipeline": ["save_pretrained", "from_pretrained"],
"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