Skip to content

GitLab

Unverified Commit 9f40d797 authored by Gang Wu's avatar Gang Wu Committed by GitHub
Browse files

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

parent a0198676
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/instruct_pix2pix/train_instruct_pix2pix_sdxl.py
View file @ 9f40d797
...@@ -103,7 +103,7 @@ def parse_args(): ...@@ -103,7 +103,7 @@ def parse_args():
) )
parser.add_argument( parser.add_argument(
"--vae_precision", "--vae_precision",
type="choice", type=str,
choices=["fp32", "fp16", "bf16"], choices=["fp32", "fp16", "bf16"],
default="fp32", default="fp32",
help=( help=(
......
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