Skip to content

Commit c60b2ed

Browse files
authored
Merge pull request #1 from Wovchena/test-stateful
Test stateful
2 parents 320fad9 + 07e3b12 commit c60b2ed

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

.github/workflows/llm_bench-python.yml

+12-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ on:
1515
branches: [ "master" ]
1616
paths:
1717
- llm_bench/python/**
18-
18+
- .github/workflows/llm_bench-python.yml
1919

2020
permissions:
2121
contents: read
@@ -56,3 +56,14 @@ jobs:
5656
with:
5757
name: llm.bench_diff
5858
path: llm.bench_diff.diff
59+
stateful:
60+
runs-on: ubuntu-20.04
61+
steps:
62+
- uses: actions/checkout@v4
63+
- uses: actions/setup-python@v4
64+
with:
65+
python-version: 3.8
66+
- name: Test stateful
67+
run: |
68+
python -m pip install -r llm_bench/python/requirements.txt optimum==1.14
69+
python llm_bench/python/convert.py --model_id TinyLlama/TinyLlama-1.1B-Chat-v0.6 --output_dir . --bettertransformer --make_stateful

0 commit comments

Comments
 (0)