Unverified Commit 9147c4c9 authored by Katsuya's avatar Katsuya Committed by GitHub
Browse files

Fix unused upcast_attn flag in convert_original_stable_diffusion_to_diffusers script (#1942)

Fix unused upcast_attn flag in sd to diffusers script
parent 6d3adf65
......@@ -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"
......
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