Skip to content

Commit 9f40d79

Browse files
authored
[FIX BUG] type of args in train_instruct_pix2pix_sdxl.py (huggingface#4955)
1 parent a019867 commit 9f40d79

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/instruct_pix2pix/train_instruct_pix2pix_sdxl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def parse_args():
103103
)
104104
parser.add_argument(
105105
"--vae_precision",
106-
type="choice",
106+
type=str,
107107
choices=["fp32", "fp16", "bf16"],
108108
default="fp32",
109109
help=(

0 commit comments

Comments
 (0)