try

huggingface · Feb 20, 2024 · f5e6b5a · f5e6b5a
1 parent 2c76d98
commit f5e6b5a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/.github/workflows/dummy.yml b/.github/workflows/dummy.yml
@@ -110,15 +110,17 @@ jobs:
 #          python3 -m pytest -v tests/models
 
 
+      # python3 src/transformers/benchmark/static_cache_benchmark.py
       # python3 src/transformers/benchmark/from_pretrained_benchmark.py
+      # git checkout best_benchmark_on_static_cache
       - name: Check transformers repository
         run: |
           git clone https://github.com/huggingface/transformers.git
           cd /transformers
           git fetch origin
-          git checkout best_benchmark_on_static_cache
+          git checkout main
           git status
           git log -n 3
           python3 -m pip uninstall -y torch torchvision torchaudio
           python3 -m pip install torch torchvision torchaudio
-          python3 src/transformers/benchmark/static_cache_benchmark.py
+          python3 static_cache_benchmark_original_1.py