diff --git a/examples/community/checkpoint_merger.py b/examples/community/checkpoint_merger.py index b6e418221ff7..764129e85900 100644 --- a/examples/community/checkpoint_merger.py +++ b/examples/community/checkpoint_merger.py @@ -5,7 +5,8 @@ import torch from diffusers import DiffusionPipeline, __version__ -from diffusers.utils import CONFIG_NAME, DIFFUSERS_CACHE, ONNX_WEIGHTS_NAME, SCHEDULER_CONFIG_NAME, WEIGHTS_NAME +from diffusers.schedulers.scheduling_utils import SCHEDULER_CONFIG_NAME +from diffusers.utils import CONFIG_NAME, DIFFUSERS_CACHE, ONNX_WEIGHTS_NAME, WEIGHTS_NAME from huggingface_hub import snapshot_download