Fix bug when converting checkpoint to diffusers format (#6900)

This fixes #6899.

Fix bug when converting checkpoint to diffusers format (#6900)
This fixes #6899.
17808a09 · Ehsan Akhgari · GitHub · 491a933a · 17808a09
Unverified Commit 17808a09 authored Feb 08, 2024 by Ehsan Akhgari Committed by GitHub Feb 08, 2024
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py ...diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +3 -0

No files found.
--- 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)