Skip to content

Commit 361c52d

Browse files
committed
fix
1 parent c47fb93 commit 361c52d

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

onnx_diagnostic/helpers/ort_session.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,9 @@ def __init__(
7777
session_options.enable_profiling = enable_profiling
7878
if optimized_model_filepath:
7979
session_options.optimized_model_filepath = optimized_model_filepath
80-
session_options.optimized_model_external_initializers_file_name = (
81-
f"{os.path.splitext(optimized_model_filepath)[0]}.data"
80+
session_options.add_session_config_entry(
81+
"session.optimized_model_external_initializers_file_name",
82+
f"{os.path.splitext(optimized_model_filepath)[0]}.data",
8283
)
8384
if log_severity_level is not None:
8485
session_options.log_severity_level = log_severity_level

onnx_diagnostic/torch_models/validate.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1771,8 +1771,9 @@ def _mk(key, flavour=flavour):
17711771
if os.environ.get("DUMPORTOPT", "") in ("1", "true", "True"):
17721772
opts = onnxruntime.SessionOptions()
17731773
opts.optimized_model_filepath = f"{data['onnx_filename']}.rtopt.onnx"
1774-
opts.optimized_model_external_initializers_file_name = (
1775-
f"{data['onnx_filename']}.rtopt.data"
1774+
opts.add_session_config_entry(
1775+
"session.optimized_model_external_initializers_file_name",
1776+
f"{data['onnx_filename']}.rtopt.data",
17761777
)
17771778
if verbose:
17781779
print(

0 commit comments

Comments
 (0)