fix nit

microsoft · Mar 22, 2024 · c5ecda3 · c5ecda3
1 parent 41328bd
commit c5ecda3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxruntime/python/tools/quantization/quantize.py b/onnxruntime/python/tools/quantization/quantize.py
@@ -659,8 +659,8 @@ def quantize(
     """Quantize a model with QuantConfig.
 
     Args:
-        model_input (Path): Path to the model or ModelProto to quantize.
-        model_output (Path): Path to save the quantized model.
+        model_input (str | Path | ModelProto): Path to the model or ModelProto to quantize.
+        model_output (str | Path): Path to save the quantized model.
         quant_config (QuantConfig): Quantization Configuration.
     """