Skip to content

Commit 00d6d1d

Browse files
committed
Add default value to 'high_precision_tensors'
Signed-off-by: gcunhase <[email protected]>
1 parent 4dbe045 commit 00d6d1d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modelopt/onnx/autocast/precisionconverter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -472,7 +472,7 @@ def _filter_unsupported_op_types(
472472
def _get_tensors_to_cast(
473473
self,
474474
low_precision_nodes: list[str],
475-
high_precision_tensors: dict[str, dict[str, list[int]]],
475+
high_precision_tensors: dict[str, dict[str, list[int]]] = {},
476476
) -> tuple[list[str], list[str], dict[str, list[onnx.NodeProto]]]:
477477
cast_to_fp16 = [] # Tensors to cast down to FP16
478478
cast_to_fp32 = [] # Tensors to cast up to FP32

0 commit comments

Comments
 (0)