Fix ONNX conversion script opset argument type (#739)

The opset argument should be an `int` but was set as a `str`.
2022-10-07 06:47:43 -07:00 · 2022-10-07 06:47:43 -07:00 · 75bb6d2d46
commit 75bb6d2d46
parent 906e4105d7
1 changed files with 1 additions and 1 deletions
--- a/scripts/convert_stable_diffusion_checkpoint_to_onnx.py
+++ b/scripts/convert_stable_diffusion_checkpoint_to_onnx.py
@ -206,7 +206,7 @@ if __name__ == "__main__":
    parser.add_argument(
        "--opset",
        default=14,
-        type=str,
+        type=int,
        help="The version of the ONNX operator set to use.",
    )