fix

huggingface · Dec 3, 2024 · e5d1045 · e5d1045
1 parent c788f82
commit e5d1045
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/test_api_rocm.yaml b/.github/workflows/test_api_rocm.yaml
@@ -32,5 +32,5 @@ jobs:
     uses: huggingface/hf-workflows/.github/workflows/optimum_benchmark_instinct_ci.yaml@main
     with:
       machine_type: single-gpu
-      install_extras: timm,diffusers,codecarbon
-      pytest_keywords: "api and cuda"
+      install_extras: testing,timm,diffusers,codecarbon
+      pytest_keywords: api and cuda
diff --git a/.github/workflows/test_cli_rocm_pytorch.yaml b/.github/workflows/test_cli_rocm_pytorch.yaml
@@ -35,7 +35,7 @@ jobs:
     with:
       machine_type: single-gpu
       install_extras: testing,diffusers,timm,peft,autoawq,auto-gptq
-      pytest_keywords: "cli and cuda and pytorch and not (dp or ddp or device_map or deepspeed) and not bnb"
+      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:
     if: ${{
@@ -52,4 +52,4 @@ jobs:
     with:
       machine_type: multi-gpu
       install_extras: testing,diffusers,timm,peft,autoawq,auto-gptq
-      pytest_keywords: "cli and cuda and pytorch and (dp or ddp or device_map)"
+      pytest_keywords: cli and cuda and pytorch and (dp or ddp or device_map)