update

Author: DN6

.github/workflows/pr_tests_gpu.yml

@@ -78,6 +78,7 @@ jobs:
         uses: actions/checkout@v3
         with:
           fetch-depth: 2
+
       - name: NVIDIA-SMI
         run: |
           nvidia-smi
@@ -86,23 +87,28 @@ jobs:
           python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
           python -m uv pip install -e [quality,test]
           pip uninstall accelerate -y && python -m uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git
+          pip uninstall transformers -y && python -m uv pip install -U transformers@git+https://github.com/huggingface/transformers.git --no-deps
+
       - name: Environment
         run: |
           python utils/print_env.py
       - name: Extract tests
+        id: extract_tests
         run: |
           pattern=$(python utils/extract_tests_from_mixin.py --type pipeline)
-          echo "TEST_PATTERN=$pattern" >> $GITHUB_ENV
+          echo "TEST_PATTERN=${pattern}" >> $GITHUB_OUTPUT
+
       - name: PyTorch CUDA checkpoint tests on Ubuntu
         env:
           HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }}
           # https://pytorch.org/docs/stable/notes/randomness.html#avoiding-nondeterministic-algorithms
           CUBLAS_WORKSPACE_CONFIG: :16:8
         run: |
           python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile \
-            -s -v -k "not Flax and not Onnx" \
+            -s -v -k "not Flax and not Onnx ${{ steps.extract_tests.outputs.TEST_PATTERN }}" \
             --make-reports=tests_pipeline_${{ matrix.module }}_cuda \
             tests/pipelines/${{ matrix.module }}
+
       - name: Failure short reports
         if: ${{ failure() }}
         run: |
@@ -149,9 +155,10 @@ jobs:
         python utils/print_env.py
 
     - name: Extract tests
+      id: extract_tests
       run: |
         pattern=$(python utils/extract_tests_from_mixin.py --type ${{ matrix.module }})
-        echo "TEST_PATTERN=$pattern" >> $GITHUB_ENV
+        echo "TEST_PATTERN=${pattern}" >> $GITHUB_OUTPUT
 
     - name: Run PyTorch CUDA tests
       env:
@@ -160,7 +167,7 @@ jobs:
         CUBLAS_WORKSPACE_CONFIG: :16:8
       run: |
         python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile \
-          -s -v -k "not Flax and not Onnx and $TEST_PATTERN" \
+          -s -v -k "not Flax and not Onnx and ${{ steps.extract_tests.outputs.TEST_PATTERN }}" \
           --make-reports=tests_torch_cuda_${{ matrix.module }} \
           tests/${{ matrix.module }}
 
@@ -179,7 +186,7 @@ jobs:
 
   run_examples_tests:
     name: Examples PyTorch CUDA tests on Ubuntu
-
+        pip uninstall transformers -y && python -m uv pip install -U transformers@git+https://github.com/huggingface/transformers.git --no-deps
     runs-on:
       group: aws-g4dn-2xlarge