[Community] Correct checkpoint merger (huggingface#1965)

AmericanPresidentJimmyCarter · Jan 10, 2023 · d1d5451 · d1d5451
1 parent f6f1ec3
commit d1d5451
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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