@@ -103,7 +103,7 @@ def parse_args():
     )
     parser.add_argument(
         "--vae_precision",
-        type="choice",
+        type=str,
         choices=["fp32", "fp16", "bf16"],
         default="fp32",
         help=(
Original file line number	Diff line number	Diff line change
`@@ -103,7 +103,7 @@ def parse_args():`
`103`	`103`	`)`
`104`	`104`	`parser.add_argument(`
`105`	`105`	`"--vae_precision",`
`106`		`- type="choice",`
	`106`	`+ type=str,`
`107`	`107`	`choices=["fp32", "fp16", "bf16"],`
`108`	`108`	`default="fp32",`
`109`	`109`	`help=(`