Update daily_for_test.yaml

zhulinJulia24 · Jan 29, 2024 · 100b48e · 100b48e
1 parent b2f2032
commit 100b48e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.github/workflows/daily_for_test.yaml b/.github/workflows/daily_for_test.yaml
@@ -11,9 +11,10 @@ env:
 jobs:
   HF_model:
     runs-on: self-hosted
+    continue-on-error: true
     strategy:
       matrix:
-        transformers-version: [4.34.0, 4.36.0, 4.37.1]
+        transformers-version: [4.34.0, 4.35.2, 4.36.2, 4.37.1]
     steps:
     - name: mask env
       run: |
@@ -26,7 +27,7 @@ jobs:
         conda create -n internlm-model-latest --clone ${CONDA_BASE_ENV}
         source activate internlm-model-latest
         # TODO:test other version of transformers
-        pip install transformers== ${{ matrix.transformers-version }}
+        pip install transformers==${{ matrix.transformers-version }}
         pip install sentencepiece
         srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=${GITHUB_RUN_ID}-${GITHUB_JOB} --gpus-per-task=2 pytest -s -v --color=yes ./tests/test_hf_model.py
         conda deactivate