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

Fix unused upcast_attn flag in convert_original_stable_diffusion_to_diffusers script (#1942)

Browse Source 
Fix unused upcast_attn flag in sd to diffusers script
This commit is contained in:
Katsuya
2023-01-12 13:55:40 -05:00
committed by GitHub
parent 6d3adf6570
commit 9147c4c954
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/convert_original_stable_diffusion_to_diffusers.py
View File

@@ -840,7 +840,7 @@ if __name__ == "__main__":
),
)
parser.add_argument(
"--upcast_attn",
"--upcast_attention",
default=False,
type=bool,
help=(
@@ -871,7 +871,7 @@ if __name__ == "__main__":
if "state_dict" in checkpoint:
checkpoint = checkpoint["state_dict"]
upcast_attention = False
upcast_attention = args.upcast_attention
if args.original_config_file is None:
key_name = "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_k.weight"