File tree Expand file tree Collapse file tree 2 files changed +5
-1
lines changed
_unittests/ut_torch_models
onnx_diagnostic/torch_models/hghub Expand file tree Collapse file tree 2 files changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -30,6 +30,7 @@ def test_get_untrained_model_with_inputs_tiny_llm(self):
3030 "input_kwargs" ,
3131 "model_kwargs" ,
3232 "task" ,
33+ "dump_info" ,
3334 },
3435 )
3536 model , inputs = data ["model" ], data ["inputs" ]
Original file line number Diff line number Diff line change @@ -135,7 +135,10 @@ def get_untrained_model_with_inputs(
135135 mkwargs [k ] = v
136136 if mkwargs :
137137 update_config (config , mkwargs )
138- diff_config = build_diff_config (config0 , config )
138+ try :
139+ diff_config = build_diff_config (config0 , config )
140+ except (ValueError , AttributeError , TypeError ) as e :
141+ diff_config = f"DIFF CONFIG ERROR { e } "
139142 if verbose :
140143 if diff_config :
141144 print ("[get_untrained_model_with_inputs] -- updated config" )
You can’t perform that action at this time.
0 commit comments