File tree Expand file tree Collapse file tree 1 file changed +0
-22
lines changed
_unittests/ut_torch_models Expand file tree Collapse file tree 1 file changed +0
-22
lines changed Original file line number Diff line number Diff line change @@ -116,28 +116,6 @@ def test_g_validate_model_onnx_dynamo_os_ort(self):
116116 onnx_filename = data ["onnx_filename" ]
117117 self .assertExists (onnx_filename )
118118
119- @requires_torch ("2.7" )
120- @hide_stdout ()
121- @ignore_warnings (FutureWarning )
122- @requires_experimental ()
123- def test_h_validate_model_custom_os_ort (self ):
124- mid = "arnir0/Tiny-LLM"
125- summary , data = validate_model (
126- mid ,
127- do_run = True ,
128- verbose = 10 ,
129- exporter = "custom" ,
130- dump_folder = "dump_test/validate_model_custom_os_ort" ,
131- patch = True ,
132- stop_if_static = 2 if pv .Version (torch .__version__ ) > pv .Version ("2.6.1" ) else 0 ,
133- optimization = "default+os_ort" ,
134- )
135- self .assertIsInstance (summary , dict )
136- self .assertIsInstance (data , dict )
137- self .assertLess (summary ["disc_onnx_ort_run_abs" ], 1e-4 )
138- onnx_filename = data ["onnx_filename" ]
139- self .assertExists (onnx_filename )
140-
141119 @requires_torch ("2.7" )
142120 @hide_stdout ()
143121 @ignore_warnings (FutureWarning )
You can’t perform that action at this time.
0 commit comments