diff --git a/.github/workflows/update_model_artifacts.yml b/.github/workflows/update_model_artifacts.yml index 8e5f871a..f7d020d4 100644 --- a/.github/workflows/update_model_artifacts.yml +++ b/.github/workflows/update_model_artifacts.yml @@ -50,7 +50,7 @@ jobs: - self-hosted # must come first - runner-group=${{ needs.setup.outputs.runner-group }} - environment=prod - - machine-type=a2-highgpu-1g + - machine-type=c2-standard-16 steps: - name: "Checking out PR repository" uses: actions/checkout@e2f20e631ae6d7dd3b768f56a5d2af784dd54791 # v2.5.0 diff --git a/common_benchmark_suite/openxla/benchmark/comparative_suite/jax/scripts/generate_model_artifacts.sh b/common_benchmark_suite/openxla/benchmark/comparative_suite/jax/scripts/generate_model_artifacts.sh index a586e151..601dd61c 100755 --- a/common_benchmark_suite/openxla/benchmark/comparative_suite/jax/scripts/generate_model_artifacts.sh +++ b/common_benchmark_suite/openxla/benchmark/comparative_suite/jax/scripts/generate_model_artifacts.sh @@ -64,9 +64,9 @@ else args+=( --filter "RESNET50_FP32_JAX_.+" - "BERT_LARGE_FP32_JAX_.+_BATCH(1|16|24|32|48|64|512)" - "T5_LARGE_FP32_JAX_.+_BATCH(1|16|24|32|48)" - "T5_4CG_LARGE_FP32_JAX_.+" + # "BERT_LARGE_FP32_JAX_.+_BATCH(1|16|24|32|48|64|512)" + # "T5_LARGE_FP32_JAX_.+_BATCH(1|16|24|32|48)" + # "T5_4CG_LARGE_FP32_JAX_.+" "GPT2LMHEAD_FP32_JAX_.+" ) fi