"src/array/vscode:/vscode.git/clone" did not exist on "9dff5419eadbe4547cbf6ac3c775f49e31715c85"
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__": ...@@ -840,7 +840,7 @@ if __name__ == "__main__":
), ),
) )
parser.add_argument( parser.add_argument(
"--upcast_attn", "--upcast_attention",
default=False, default=False,
type=bool, type=bool,
help=( help=(
...@@ -871,7 +871,7 @@ if __name__ == "__main__": ...@@ -871,7 +871,7 @@ if __name__ == "__main__":
if "state_dict" in checkpoint: if "state_dict" in checkpoint:
checkpoint = checkpoint["state_dict"] checkpoint = checkpoint["state_dict"]
upcast_attention = False upcast_attention = args.upcast_attention
if args.original_config_file is None: if args.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"
......
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