|
20 | 20 | CUSTOM_RUNNERS = { |
21 | 21 | "linux": { |
22 | 22 | # This one runs OOM on smaller runner, the root cause is unclear (T163016365) |
23 | | - "w2l": "linux.12xlarge", |
24 | | - "ic4": "linux.12xlarge", |
25 | | - "resnet50": "linux.12xlarge", |
26 | | - "llava": "linux.12xlarge", |
27 | | - "llama3_2_vision_encoder": "linux.12xlarge", |
28 | | - "llama3_2_text_decoder": "linux.12xlarge", |
| 23 | + "w2l": "linux.4xlarge.memory", |
| 24 | + "ic4": "linux.4xlarge.memory", |
| 25 | + "resnet50": "linux.4xlarge.memory", |
| 26 | + "llava": "linux.4xlarge.memory", |
| 27 | + "llama3_2_vision_encoder": "linux.4xlarge.memory", |
| 28 | + "llama3_2_text_decoder": "linux.4xlarge.memory", |
29 | 29 | # This one causes timeout on smaller runner, the root cause is unclear (T161064121) |
30 | | - "dl3": "linux.12xlarge", |
31 | | - "emformer_join": "linux.12xlarge", |
32 | | - "emformer_predict": "linux.12xlarge", |
| 30 | + "dl3": "linux.4xlarge.memory", |
| 31 | + "emformer_join": "linux.4xlarge.memory", |
| 32 | + "emformer_predict": "linux.4xlarge.memory", |
33 | 33 | } |
34 | 34 | } |
35 | 35 |
|
|
0 commit comments