From 01fecd1000a07dc663d2c8e3185ebc379cc823d8 Mon Sep 17 00:00:00 2001 From: Guang Yang Date: Mon, 22 Apr 2024 13:17:40 -0700 Subject: [PATCH] [RELEASE ONLY] Fix running llava_encoder in CI --- .ci/scripts/gather_test_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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",