File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
tests/models/testing_utils Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -375,7 +375,7 @@ def _check_model_hotswap(
375375 # additionally check if dynamic compilation works.
376376 if different_shapes is not None :
377377 for height , width in different_shapes :
378- new_inputs_dict = self .prepare_dummy_input (height = height , width = width )
378+ new_inputs_dict = self .get_dummy_inputs (height = height , width = width )
379379 _ = model (** new_inputs_dict )
380380 else :
381381 output0_after = model (** inputs_dict )["sample" ]
@@ -390,7 +390,7 @@ def _check_model_hotswap(
390390 with torch .inference_mode ():
391391 if different_shapes is not None :
392392 for height , width in different_shapes :
393- new_inputs_dict = self .prepare_dummy_input (height = height , width = width )
393+ new_inputs_dict = self .get_dummy_inputs (height = height , width = width )
394394 _ = model (** new_inputs_dict )
395395 else :
396396 output1_after = model (** inputs_dict )["sample" ]
You can’t perform that action at this time.
0 commit comments