This commit is contained in:
ydshieh 2024-05-30 16:59:59 +02:00
parent 8c2a653809
commit 100e0dc336
1 changed files with 1 additions and 1 deletions

View File

@ -37,4 +37,4 @@ jobs:
run: |
echo "Hello Benchmark"
python3 -m pip install optimum-benchmark>=0.2.0
python3 benchmark/benchmark.py --config-dir benchmark/config --config-name generation --commit=${{ github.sha }} backend.model=google/gemma-2b backend.cache_implementation=null,static backend.torch_compile=false,true --multirun
HF_TOKEN=${{ secrets.HF_HUB_READ_TOKEN }} python3 benchmark/benchmark.py --config-dir benchmark/config --config-name generation --commit=${{ github.sha }} backend.model=google/gemma-2b backend.cache_implementation=null,static backend.torch_compile=false,true --multirun