Fix ONNX conversion script opset argument type (#739)

justinchuby · web-flow · commit 75bb6d2d466d · 2022-10-07T15:47:43.000+02:00
The opset argument should be an `int` but was set as a `str`.
diff --git a/scripts/convert_stable_diffusion_checkpoint_to_onnx.py b/scripts/convert_stable_diffusion_checkpoint_to_onnx.py
@@ -206,7 +206,7 @@ def convert_models(model_path: str, output_path: str, opset: int):
     parser.add_argument(
         "--opset",
         default=14,
-        type=str,
+        type=int,
         help="The version of the ONNX operator set to use.",
     )
 

Original file line number	Diff line number	Diff line change
`@@ -206,7 +206,7 @@ def convert_models(model_path: str, output_path: str, opset: int):`
`206`	`206`	`parser.add_argument(`
`207`	`207`	`"--opset",`
`208`	`208`	`default=14,`
`209`		`- type=str,`
	`209`	`+ type=int,`
`210`	`210`	`help="The version of the ONNX operator set to use.",`
`211`	`211`	`)`
`212`	`212`