Skip to content

Commit

Permalink
Update convert_from_ckpt.py / read checkpoint config yaml contents (#…
Browse files Browse the repository at this point in the history
…6633)

Update convert_from_ckpt.py / read config yaml contents

Added missing reading of config yaml file contents
  • Loading branch information
spezialspezial authored Jan 19, 2024
1 parent a65ca8a commit 58b8dce
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down

0 comments on commit 58b8dce

Please sign in to comment.