[FIX BUG] type of args in train_instruct_pix2pix_sdxl.py (huggingface…

…#4955)
kashif · Sep 16, 2023 · 69e5f6c · 69e5f6c
1 parent e0659f6
commit 69e5f6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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=(