|
87 | 87 | python utils/print_env.py |
88 | 88 | - name: Slow PyTorch CUDA checkpoint tests on Ubuntu |
89 | 89 | env: |
90 | | - HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }} |
| 90 | + HF_TOKEN: ${{ secrets.HF_TOKEN }} |
91 | 91 | # https://pytorch.org/docs/stable/notes/randomness.html#avoiding-nondeterministic-algorithms |
92 | 92 | CUBLAS_WORKSPACE_CONFIG: :16:8 |
93 | 93 | run: | |
@@ -144,7 +144,7 @@ jobs: |
144 | 144 |
|
145 | 145 | - name: Run slow PyTorch CUDA tests |
146 | 146 | env: |
147 | | - HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }} |
| 147 | + HF_TOKEN: ${{ secrets.HF_TOKEN }} |
148 | 148 | # https://pytorch.org/docs/stable/notes/randomness.html#avoiding-nondeterministic-algorithms |
149 | 149 | CUBLAS_WORKSPACE_CONFIG: :16:8 |
150 | 150 | run: | |
@@ -194,7 +194,7 @@ jobs: |
194 | 194 |
|
195 | 195 | - name: Run slow PEFT CUDA tests |
196 | 196 | env: |
197 | | - HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }} |
| 197 | + HF_TOKEN: ${{ secrets.HF_TOKEN }} |
198 | 198 | # https://pytorch.org/docs/stable/notes/randomness.html#avoiding-nondeterministic-algorithms |
199 | 199 | CUBLAS_WORKSPACE_CONFIG: :16:8 |
200 | 200 | run: | |
@@ -243,7 +243,7 @@ jobs: |
243 | 243 |
|
244 | 244 | - name: Run slow Flax TPU tests |
245 | 245 | env: |
246 | | - HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }} |
| 246 | + HF_TOKEN: ${{ secrets.HF_TOKEN }} |
247 | 247 | run: | |
248 | 248 | python -m pytest -n 0 \ |
249 | 249 | -s -v -k "Flax" \ |
@@ -290,7 +290,7 @@ jobs: |
290 | 290 |
|
291 | 291 | - name: Run slow ONNXRuntime CUDA tests |
292 | 292 | env: |
293 | | - HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }} |
| 293 | + HF_TOKEN: ${{ secrets.HF_TOKEN }} |
294 | 294 | run: | |
295 | 295 | python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile \ |
296 | 296 | -s -v -k "Onnx" \ |
@@ -337,7 +337,7 @@ jobs: |
337 | 337 | python utils/print_env.py |
338 | 338 | - name: Run example tests on GPU |
339 | 339 | env: |
340 | | - HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }} |
| 340 | + HF_TOKEN: ${{ secrets.HF_TOKEN }} |
341 | 341 | run: | |
342 | 342 | python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile -s -v -k "compile" --make-reports=tests_torch_compile_cuda tests/ |
343 | 343 | - name: Failure short reports |
@@ -378,7 +378,7 @@ jobs: |
378 | 378 | python utils/print_env.py |
379 | 379 | - name: Run example tests on GPU |
380 | 380 | env: |
381 | | - HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }} |
| 381 | + HF_TOKEN: ${{ secrets.HF_TOKEN }} |
382 | 382 | run: | |
383 | 383 | python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile -s -v -k "xformers" --make-reports=tests_torch_xformers_cuda tests/ |
384 | 384 | - name: Failure short reports |
@@ -423,7 +423,7 @@ jobs: |
423 | 423 |
|
424 | 424 | - name: Run example tests on GPU |
425 | 425 | env: |
426 | | - HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }} |
| 426 | + HF_TOKEN: ${{ secrets.HF_TOKEN }} |
427 | 427 | run: | |
428 | 428 | python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH" |
429 | 429 | python -m uv pip install timm |
|
0 commit comments