Update on "Add nightly model tests against pytorch"

[ghstack-poisoned]
pytorch · May 29, 2024 · 1a61b58 · 1a61b58
1 parent 7401331
commit 1a61b58
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/.github/workflows/model_tests.yaml b/.github/workflows/model_tests.yaml
@@ -55,6 +55,8 @@ jobs:
         run: torchrun --nproc-per-node 4 examples/huggingface/pippy_convBert.py
       - name: Run deberta
         run: torchrun --nproc-per-node 4 examples/huggingface/pippy_deberta.py
+      # - name: Run debertaV2
+      #   run: torchrun --nproc-per-node 4 examples/huggingface/pippy_debertaV2.py
       - name: Run distilBert
         run: torchrun --nproc-per-node 4 examples/huggingface/pippy_distilBert.py
       - name: Run electra
@@ -71,8 +73,8 @@ jobs:
         run: torchrun --nproc-per-node 4 examples/huggingface/pippy_megatronBert.py
       - name: Run mobileBert
         run: torchrun --nproc-per-node 4 examples/huggingface/pippy_mobileBert.py
-      - name: Run opt
-        run: torchrun --nproc-per-node 2 examples/huggingface/pippy_opt.py
+      # - name: Run opt
+      #   run: torchrun --nproc-per-node 2 examples/huggingface/pippy_opt.py
       - name: Run trOCR
         run: torchrun --nproc-per-node 4 examples/huggingface/pippy_trOCR.py
       - name: Run unet