token

Author: hlky

.github/workflows/pr_tests_mps.yml

@@ -97,52 +97,66 @@ jobs:
 
     - name: Run fast PyTorch Pipeline MPS tests
       if: ${{ matrix.config.framework == 'pytorch_pipelines' }}
+      env:
+        HF_TOKEN: ${{ secrets.HF_TOKEN }}
       run: |
         ${CONDA_RUN} python -m pytest -n 0 -s -v -k "not Flax and not Onnx" \
         --make-reports=tests_${{ matrix.config.report }} \
           tests/pipelines
 
     - name: Run fast PyTorch Models MPS tests
       if: ${{ matrix.config.framework == 'pytorch_models' }}
+      env:
+        HF_TOKEN: ${{ secrets.HF_TOKEN }}
       run: |
         ${CONDA_RUN} python -m pytest -n 0 -s -v -k "not Flax and not Onnx and not Dependency" \
         --make-reports=tests_${{ matrix.config.report }} \
           tests/models/
 
     - name: Run fast PyTorch Schedulers MPS tests
       if: ${{ matrix.config.framework == 'pytorch_schedulers' }}
+      env:
+        HF_TOKEN: ${{ secrets.HF_TOKEN }}
       run: |
         ${CONDA_RUN} python -m pytest -n 0 -s -v -k "not Flax and not Onnx and not Dependency" \
         --make-reports=tests_${{ matrix.config.report }} \
           tests/schedulers/
 
     - name: Run fast PyTorch Others MPS tests
       if: ${{ matrix.config.framework == 'pytorch_others' }}
+      env:
+        HF_TOKEN: ${{ secrets.HF_TOKEN }}
       run: |
         ${CONDA_RUN} python -m pytest -n 0 -s -v \
         --make-reports=tests_${{ matrix.config.report }} \
           tests/others/
 
     - name: Run fast PyTorch Single File MPS tests
       if: ${{ matrix.config.framework == 'pytorch_single_file' }}
+      env:
+        HF_TOKEN: ${{ secrets.HF_TOKEN }}
       run: |
         ${CONDA_RUN} python -m pytest -n 0 -s -v \
         --make-reports=tests_${{ matrix.config.report }} \
           tests/single_file/
 
     - name: Run fast PyTorch Lora MPS tests
       if: ${{ matrix.config.framework == 'pytorch_lora' }}
+      env:
+        HF_TOKEN: ${{ secrets.HF_TOKEN }}
       run: |
         ${CONDA_RUN} python -m pytest -n 0 -s -v \
         --make-reports=tests_${{ matrix.config.report }} \
           tests/lora/
 
-    - name: Run fast PyTorch Quantization MPS tests
-      if: ${{ matrix.config.framework == 'pytorch_quantization' }}
-      run: |
-        ${CONDA_RUN} python -m pytest -n 0 -s -v \
-        --make-reports=tests_${{ matrix.config.report }} \
-          tests/quantization/
+    # - name: Run fast PyTorch Quantization MPS tests
+    #   if: ${{ matrix.config.framework == 'pytorch_quantization' }}
+    #   env:
+    #     HF_TOKEN: ${{ secrets.HF_TOKEN }}
+    #   run: |
+    #     ${CONDA_RUN} python -m pytest -n 0 -s -v \
+    #     --make-reports=tests_${{ matrix.config.report }} \
+    #       tests/quantization/
 
     - name: Failure short reports
       if: ${{ failure() }}