From d28451007bff5bccbfd8f66cfec03d9b5ed7237b Mon Sep 17 00:00:00 2001 From: Justin Chu Date: Wed, 5 Oct 2022 08:56:16 -0700 Subject: [PATCH] Fix ONNX conversion script opset argument type The opset argument should be an `int` but was set as a `str`. --- scripts/convert_stable_diffusion_checkpoint_to_onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/convert_stable_diffusion_checkpoint_to_onnx.py b/scripts/convert_stable_diffusion_checkpoint_to_onnx.py index 07dc21aa3b16..a388c0d078dc 100644 --- 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.", )