From 58b8dce129dc108acf142f11ae514328e37516a3 Mon Sep 17 00:00:00 2001 From: spezialspezial <75758219+spezialspezial@users.noreply.github.com> Date: Fri, 19 Jan 2024 03:55:03 +0100 Subject: [PATCH] Update convert_from_ckpt.py / read checkpoint config yaml contents (#6633) Update convert_from_ckpt.py / read config yaml contents Added missing reading of config yaml file contents --- src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py index b871f7bbe5..1ba06f811a 100644 --- a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +++ b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py @@ -1317,6 +1317,9 @@ def download_from_original_stable_diffusion_ckpt( if config_url is not None: original_config_file = BytesIO(requests.get(config_url).content) + else: + with open(original_config_file, "r") as f: + original_config_file = f.read() original_config = yaml.safe_load(original_config_file)