From f5e6b5a214fe9eb0689895b80dff8c3a21d605d9 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Tue, 20 Feb 2024 14:24:50 +0800 Subject: [PATCH] try --- .github/workflows/dummy.yml | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/.github/workflows/dummy.yml b/.github/workflows/dummy.yml index df07c585..4665af3f 100644 --- 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 \ No newline at end of file + python3 static_cache_benchmark_original_1.py