Add model accuracy test - step 1 (#866)
This commit is contained in:
2
.github/workflows/unit-test.yml
vendored
2
.github/workflows/unit-test.yml
vendored
@@ -35,6 +35,7 @@ jobs:
|
||||
pip install -e "python[all]"
|
||||
pip install flashinfer -i https://flashinfer.ai/whl/cu121/torch2.3/ --force-reinstall
|
||||
pip install --upgrade transformers
|
||||
pip install accelerate
|
||||
|
||||
- name: Test Frontend Language with SRT Backend
|
||||
run: |
|
||||
@@ -50,6 +51,7 @@ jobs:
|
||||
run: |
|
||||
cd test/srt
|
||||
python3 test_eval_accuracy.py
|
||||
python3 models/test_causal_models.py
|
||||
|
||||
- name: Test Frontend Language with OpenAI Backend
|
||||
run: |
|
||||
|
||||
Reference in New Issue
Block a user