Skip to content

Commit 171b78a

Browse files
committed
nit: updated comment
Signed-off-by: gcunhase <[email protected]>
1 parent 69cb674 commit 171b78a

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
@@ -171,9 +171,8 @@ def get_parser() -> argparse.ArgumentParser:
171171
"A space-separated list indicating the precision for each custom op. "
172172
"Each item should have the format <op_type>:<precision> (all inputs and outputs have the same precision) "
173173
"or <op_type>:[<inp1_precision>,<inp2_precision>,...]:[<out1_precision>,<out2_precision>,...] "
174-
"(inputs and outputs can have different precisions), where precision can be fp32 (default), "
175-
"fp16, int8, or fp8. Note that int8/fp8 should be the same as the quantization mode. "
176-
"For example: op_type_1:fp16 op_type_2:[int8,fp32]:[int8]."
174+
"(inputs and outputs can have different precisions), where precision can be fp32 (default) or fp16."
175+
"For example: op_type_1:fp16 op_type_2:[fp16,fp32]:[fp16]."
177176
),
178177
)
179178

0 commit comments

Comments
 (0)