diff --git a/.ci/scripts/gather_test_models.py b/.ci/scripts/gather_test_models.py index ef65b6f9b42..717eff6157d 100755 --- a/.ci/scripts/gather_test_models.py +++ b/.ci/scripts/gather_test_models.py @@ -23,7 +23,7 @@ "w2l": "linux.12xlarge", "ic4": "linux.12xlarge", "resnet50": "linux.12xlarge", - "llava_encoder": "linux.4xlarge", + "llava_encoder": "linux.12xlarge", # This one causes timeout on smaller runner, the root cause is unclear (T161064121) "dl3": "linux.12xlarge", "emformer_join": "linux.12xlarge",