AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Update convert_from_ckpt.py / read checkpoint config yaml contents (#6633)

Browse Source 
Update convert_from_ckpt.py / read config yaml contents

Added missing reading of config yaml file contents
This commit is contained in:
spezialspezial
2024-01-19 03:55:03 +01:00
committed by GitHub
parent a65ca8a059
commit 58b8dce129
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py
View File

@@ -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)