"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "1ec28a2c770546e4483a5109c4e98d6e1b298968"
Unverified Commit 0850b88f authored by Plat's avatar Plat Committed by GitHub
Browse files

Fix typo in load_pipeline_from_original_stable_diffusion_ckpt() method (#2320)

fix typo
parent 5d4f59ee
...@@ -871,7 +871,7 @@ def load_pipeline_from_original_stable_diffusion_ckpt( ...@@ -871,7 +871,7 @@ def load_pipeline_from_original_stable_diffusion_ckpt(
if original_config_file is None: if original_config_file is None:
key_name = "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_k.weight" key_name = "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_k.weight"
original_config_file = os.path.join(tmpdir, "inferenc.yaml") original_config_file = os.path.join(tmpdir, "inference.yaml")
if key_name in checkpoint and checkpoint[key_name].shape[-1] == 1024: if key_name in checkpoint and checkpoint[key_name].shape[-1] == 1024:
if not os.path.isfile("v2-inference-v.yaml"): if not os.path.isfile("v2-inference-v.yaml"):
# model_type = "v2" # model_type = "v2"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment