From 6053e6dcb24287f47f0c50ebfbdfeda97e22afa0 Mon Sep 17 00:00:00 2001 From: Aleks Knezevic Date: Sat, 21 Dec 2024 15:56:04 +0000 Subject: [PATCH] Back down to 3 jobs, allow execution on any runer --- .github/workflows/run-model-tests.yml | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/.github/workflows/run-model-tests.yml b/.github/workflows/run-model-tests.yml index 1d37db0..2cdde77 100644 --- a/.github/workflows/run-model-tests.yml +++ b/.github/workflows/run-model-tests.yml @@ -14,10 +14,9 @@ jobs: fail-fast: false matrix: build: [ - {runs-on: n150, n300, name: "run1", test_names: "stable_diffusion, Qwen, MobileNetV2, clip, flan_t5, mlpmixer, resnet, vilt, albert"}, - {runs-on: n150, n300, name: "run2", test_names: "codegen, glpn_kitti, mnist, resnet50, t5, whisper, autoencoder_conv, deit, gpt2, mobilenet_ssd, roberta, timm, xglm, autoencoder_linear, detr, beit, distilbert, hand_landmark, openpose, segment_anything, unet, yolov3, bert, dpr, hardnet, opt"}, - {runs-on: n300, n150, name: "run3", test_names: "speecht5_tts, unet_brain, yolov5, bloom, falcon, llama, perceiver_io, squeeze_bert, unet_carvana, mgp-str-base, musicgen_small, segformer, torchvision, yolos"}, - {runs-on: n300, n150, name: "run4", test_names: "RMBG, gpt_neo"}, + {runs-on: runner, n300, name: "run1", test_names: "stable_diffusion, Qwen, MobileNetV2, clip, flan_t5, mlpmixer, resnet, vilt, albert, codegen, glpn_kitti, mnist, resnet50"}, + {runs-on: runner, n300, name: "run2", test_names: "t5, whisper, autoencoder_conv, deit, gpt2, mobilenet_ssd, roberta, timm, xglm, autoencoder_linear, detr, beit, distilbert, hand_landmark, openpose, segment_anything, unet, yolov3, bert, dpr, hardnet, opt, speecht5_tts, unet_brain, yolov5, bloom, falcon, llama, perceiver_io, squeeze_bert"}, + {runs-on: runner, n150, name: "run3", test_names: "RMBG, gpt_neo, unet_carvana, mgp-str-base, musicgen_small, segformer, torchvision, yolos"}, ] runs-on: - ${{ matrix.build.runs-on }}