Skip to content

Commit dfd7d55

Browse files
committed
nit: updated comment
Signed-off-by: gcunhase <[email protected]>
1 parent 7fef5a5 commit dfd7d55

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

modelopt/onnx/autocast/__main__.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -152,9 +152,8 @@ def get_parser() -> argparse.ArgumentParser:
152152
"A space-separated list indicating the precision for each custom op. "
153153
"Each item should have the format <op_type>:<precision> (all inputs and outputs have the same precision) "
154154
"or <op_type>:[<inp1_precision>,<inp2_precision>,...]:[<out1_precision>,<out2_precision>,...] "
155-
"(inputs and outputs can have different precisions), where precision can be fp32 (default), "
156-
"fp16, int8, or fp8. Note that int8/fp8 should be the same as the quantization mode. "
157-
"For example: op_type_1:fp16 op_type_2:[int8,fp32]:[int8]."
155+
"(inputs and outputs can have different precisions), where precision can be fp32 (default) or fp16."
156+
"For example: op_type_1:fp16 op_type_2:[fp16,fp32]:[fp16]."
158157
),
159158
)
160159

0 commit comments

Comments
 (0)