[FIX BUG] type of args in train_instruct_pix2pix_sdxl.py (#4955)

9f40d797 · Gang Wu · GitHub · a0198676 · 9f40d797
Unverified Commit 9f40d797 authored Sep 15, 2023 by Gang Wu Committed by GitHub Sep 15, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/instruct_pix2pix/train_instruct_pix2pix_sdxl.py examples/instruct_pix2pix/train_instruct_pix2pix_sdxl.py +1 -1

No files found.
--- a/examples/instruct_pix2pix/train_instruct_pix2pix_sdxl.py
+++ b/examples/instruct_pix2pix/train_instruct_pix2pix_sdxl.py
@@ -103,7 +103,7 @@ def parse_args():
    )
    parser.add_argument(
        "--vae_precision",
-        type="choice",
+        type=str,
        choices=["fp32", "fp16", "bf16"],
        default="fp32",
        help=(