From d1d5451b64fd1bdacc70a251269267e41d87f1ea Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 10 Jan 2023 15:11:50 +0100 Subject: [PATCH] [Community] Correct checkpoint merger (#1965) --- examples/community/checkpoint_merger.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/community/checkpoint_merger.py b/examples/community/checkpoint_merger.py index b6e418221f..764129e859 100644 --- a/examples/community/checkpoint_merger.py +++ b/examples/community/checkpoint_merger.py @@ -5,7 +5,8 @@ from typing import Dict, List, Union 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