Static cache in prefill, static batch size #164
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: Integration test | |
on: [push] | |
env: | |
TORCH_DEVICE: "cpu" | |
jobs: | |
build: | |
runs-on: ubuntu-latest | |
steps: | |
- uses: actions/checkout@v3 | |
- name: Set up Python 3.11 | |
uses: actions/setup-python@v4 | |
with: | |
python-version: 3.11 | |
- name: Install apt dependencies | |
run: | | |
sudo apt-get update | |
sudo apt-get install -y tesseract-ocr tesseract-ocr-eng | |
- name: Install python dependencies | |
run: | | |
pip install poetry | |
poetry install | |
poetry remove torch | |
poetry run pip install torch --index-url https://download.pytorch.org/whl/cpu | |
- name: Run detection benchmark test | |
run: | | |
poetry run python benchmark/detection.py --max 2 | |
poetry run python scripts/verify_benchmark_scores.py results/benchmark/det_bench/results.json --bench_type detection | |
- name: Run recognition benchmark test | |
run: | | |
poetry run python benchmark/recognition.py --max 2 | |
poetry run python scripts/verify_benchmark_scores.py results/benchmark/rec_bench/results.json --bench_type recognition | |
- name: Run layout benchmark test | |
run: | | |
poetry run python benchmark/layout.py --max 5 | |
poetry run python scripts/verify_benchmark_scores.py results/benchmark/layout_bench/results.json --bench_type layout | |
- name: Run ordering benchmark text | |
run: | | |
poetry run python benchmark/ordering.py --max 5 | |
poetry run python scripts/verify_benchmark_scores.py results/benchmark/order_bench/results.json --bench_type ordering | |