Skip to content

Commit 4f3ee41

Browse files
galagamgcunhase
andcommitted
Update docstring
Co-authored-by: Gwena Cunha <4861122+gcunhase@users.noreply.github.com> Signed-off-by: Gal Hubara-Agam <96368689+galagam@users.noreply.github.com>
1 parent 132ac9d commit 4f3ee41

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modelopt/onnx/autocast/convert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,7 @@ def convert_to_f16(
220220
infer_shapes. This is a workaround (WAR) when only type inference is
221221
needed without shape inference. Default: False.
222222
opset: Target ONNX opset version. If None, uses default minimum opset based on precision type
223-
(22 for bf16, 13 for fp16) and Q/DQ node requirements. The opset may be automatically
223+
(22 for bf16, 19 for fp16) and Q/DQ node requirements. The opset may be automatically
224224
increased if Q/DQ nodes in the model require a higher version (e.g., FP8 requires 19,
225225
INT4 requires 21, NVFP4 requires 23).
226226
"""

0 commit comments

Comments
 (0)