Fix ONNX conversion script opset argument type (#739)

The opset argument should be an `int` but was set as a `str`.

Fix ONNX conversion script opset argument type (#739)
The opset argument should be an `int` but was set as a `str`.
75bb6d2d · Justin Chu · GitHub · 906e4105 · 75bb6d2d
Unverified Commit 75bb6d2d authored Oct 07, 2022 by Justin Chu Committed by GitHub Oct 07, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

scripts/convert_stable_diffusion_checkpoint_to_onnx.py scripts/convert_stable_diffusion_checkpoint_to_onnx.py +1 -1

No files found.
--- 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.",
    )