From e91e469cd7412ebc414b6df79bb52281622875d2 Mon Sep 17 00:00:00 2001 From: Ehsan Akhgari Date: Wed, 7 Feb 2024 18:40:47 -0500 Subject: [PATCH] Fix bug when converting checkpoint to diffusers format This fixes #6899. --- 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 1ba06f811a02..f966f9794d3b 100644 --- a/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +++ b/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py @@ -1320,6 +1320,9 @@ def download_from_original_stable_diffusion_ckpt( else: with open(original_config_file, "r") as f: original_config_file = f.read() + else: + with open(original_config_file, "r") as f: + original_config_file = f.read() original_config = yaml.safe_load(original_config_file)