Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
IlyasMoutawwakil committed Dec 11, 2024
1 parent 7ed0776 commit 18db6fd
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 7 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/test_api_rocm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ jobs:
with:
machine_type: single-gpu
install_extras: testing,timm,diffusers,codecarbon
test_file: tests/test_api.py
test_file: test_api.py
pytest_keywords: api and cuda
secrets:
HF_TOKEN: ${{ secrets.HF_TOKEN }}
7 changes: 3 additions & 4 deletions .github/workflows/test_cli_cuda_tensorrt_llm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ jobs:

- name: Install dependencies
run: |
pip install -e .[testing]
pip install -e .[testing,tesnsorrt-llm]
- name: Run tests
run: |
Expand All @@ -57,8 +57,7 @@ jobs:
}}
name: Run examples
run: |
pip install huggingface-cli[cli]
huggingface-cli delete-cache
rm -rf /root/.cache/huggingface
pytest tests/test_examples.py -x -s -k "cli and cuda and trt"
cli_cuda_tensorrt_llm_multi_gpu_tests:
Expand All @@ -85,7 +84,7 @@ jobs:

- name: Install dependencies
run: |
pip install -e .[testing]
pip install -e .[testing,tesnsorrt-llm]
- name: Run tests (sequential)
run: |
Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/test_cli_rocm_pytorch.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -34,8 +34,8 @@ jobs:
uses: huggingface/hf-workflows/.github/workflows/optimum_benchmark_instinct_ci.yaml@testing
with:
machine_type: single-gpu
test_file: tests/test_cli.py
install_extras: testing,diffusers,timm,peft,autoawq,auto-gptq
test_file: test_cli.py
pytest_keywords: cli and cuda and pytorch and not (dp or ddp or device_map or deepspeed) and not bnb

run_cli_rocm_pytorch_multi_gpu_tests:
Expand All @@ -52,6 +52,6 @@ jobs:
uses: huggingface/hf-workflows/.github/workflows/optimum_benchmark_instinct_ci.yaml@testing
with:
machine_type: multi-gpu
test_file: tests/test_cli.py
install_extras: testing,diffusers,timm,peft
test_file: test_cli.py
pytest_keywords: cli and cuda and pytorch and (dp or ddp or device_map)

0 comments on commit 18db6fd

Please sign in to comment.