File tree Expand file tree Collapse file tree 2 files changed +5
-5
lines changed
onnx_diagnostic/torch_models Expand file tree Collapse file tree 2 files changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -77,10 +77,10 @@ the same model.
7777
7878.. code-block ::
7979
80- python -m onnx_diagnostic validate -m arnir0/Tiny-LLM --run -v 1 --export exporter -nostrict -o dump_models --patch
80+ python -m onnx_diagnostic validate -m arnir0/Tiny-LLM --run -v 1 --export export -nostrict -o dump_models --patch
8181
8282 .. runpython ::
8383
8484 from onnx_diagnostic._command_lines_parser import main
8585
86- main("validate -m arnir0/Tiny-LLM --run -v 1 --export exporter -nostrict -o dump_models --patch".split())
86+ main("validate -m arnir0/Tiny-LLM --run -v 1 --export export -nostrict -o dump_models --patch".split())
Original file line number Diff line number Diff line change @@ -168,7 +168,7 @@ def validate_model(
168168 assert hash_inputs == string_type (data ["inputs" ], with_shape = True ), (
169169 f"The model execution did modified the inputs:\n "
170170 f"before: { hash_inputs } \n "
171- f" after: { string_type (data [" inputs" ], with_shape = True )} "
171+ f" after: { string_type (data [' inputs' ], with_shape = True )} "
172172 )
173173
174174 if exporter :
@@ -220,7 +220,7 @@ def validate_model(
220220 ), (
221221 f"The model execution did modified the inputs:\n "
222222 f"before: { hash_inputs } \n "
223- f" after: { string_type (data [" inputs_export" ], with_shape = True )} "
223+ f" after: { string_type (data [' inputs_export' ], with_shape = True )} "
224224 )
225225
226226 # data is modified inplace
@@ -425,6 +425,6 @@ def call_torch_export_export(
425425 assert hash_inputs == string_type (data ["inputs_export" ], with_shape = True ), (
426426 f"The exported model execution did modified the inputs:\n "
427427 f"before: { hash_inputs } \n "
428- f" after: { string_type (data [" inputs_export" ], with_shape = True )} "
428+ f" after: { string_type (data [' inputs_export' ], with_shape = True )} "
429429 )
430430 return summary , data
You can’t perform that action at this time.
0 commit comments