Skip to content

Commit d4eee50

Browse files
scripts/vsmlrt.py: switch modelopt converter
1 parent 0f5da54 commit d4eee50

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

scripts/vsmlrt.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
__version__ = "3.22.29"
1+
__version__ = "3.22.30"
22

33
__all__ = [
44
"Backend", "BackendV2",
@@ -2388,8 +2388,8 @@ def tensorrt_rtx(
23882388
if not os.access(fp16_network_path, mode=os.R_OK):
23892389
import onnx
23902390
try:
2391-
from modelopt.onnx.autocast import convert_to_mixed_precision
2392-
model = convert_to_mixed_precision(network_path, keep_io_types=not fp16_io)
2391+
from modelopt.onnx.autocast import convert_to_f16
2392+
model = convert_to_f16(network_path, keep_io_types=not fp16_io)
23932393
except Exception:
23942394
from onnxconverter_common.float16 import convert_float_to_float16
23952395
model = onnx.load(network_path)

0 commit comments

Comments
 (0)