From 17808a091e2d5615c2ed8a63d7ae6f2baea11e1e Mon Sep 17 00:00:00 2001 From: Ehsan Akhgari Date: Thu, 8 Feb 2024 08:22:11 -0500 Subject: [PATCH] Fix bug when converting checkpoint to diffusers format (#6900) 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 1ba06f811a..f966f9794d 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)