File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed
_unittests/ut_torch_models Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -259,18 +259,18 @@ def test_validate_model_vit_model(self):
259259 @hide_stdout ()
260260 @ignore_warnings (FutureWarning )
261261 @requires_transformers ("4.51" )
262- def test_validate_phi3_mini_4k_instruct (self ):
263- mid = "microsoft/Phi-3-mini-4k -instruct"
262+ def test_validate_phi35_mini_instruct (self ):
263+ mid = "microsoft/Phi-3.5 -mini-instruct"
264264 summary , data = validate_model (
265265 mid ,
266266 do_run = True ,
267267 verbose = 10 ,
268- exporter = "onnx-dynamo " ,
269- dump_folder = "dump_test/validate_phi3_mini_4k_instruct " ,
268+ exporter = "custom " ,
269+ dump_folder = "dump_test/validate_phi35_mini_instruct " ,
270270 inputs2 = True ,
271271 patch = True ,
272272 rewrite = True ,
273- model_options = {"rope_scaling" : {"rope_type" : "dynamic" , "factor" : 10.0 }},
273+ # model_options={"rope_scaling": {"rope_type": "dynamic", "factor": 10.0}},
274274 )
275275 self .assertIsInstance (summary , dict )
276276 self .assertIsInstance (data , dict )
You can’t perform that action at this time.
0 commit comments