Fix CI OOM issue

pytorch · Aug 1, 2024 · c328b01 · c328b01
1 parent ad371a4
commit c328b01
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.ci/scripts/gather_test_models.py b/.ci/scripts/gather_test_models.py
@@ -27,6 +27,7 @@
         # This one causes timeout on smaller runner, the root cause is unclear (T161064121)
         "dl3": "linux.12xlarge",
         "emformer_join": "linux.12xlarge",
+        "emformer_predict": "linux.12xlarge",
     }
 }