Skip to content

Commit db2d7e7

Browse files
DN6sayakpaul
andauthored
[CI] Fix new LoRAHotswap tests (#13163)
update Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
1 parent f8d3db9 commit db2d7e7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/models/testing_utils/lora.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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"]

0 commit comments

Comments
 (0)