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