mirror of
https://github.com/VikParuchuri/surya.git
synced 2026-06-04 21:03:53 +08:00
41 lines
1.6 KiB
YAML
41 lines
1.6 KiB
YAML
name: Test CLI scripts
|
|
|
|
on: [push]
|
|
|
|
jobs:
|
|
build:
|
|
runs-on: t4_gpu
|
|
env:
|
|
# T4 (Turing, compute 7.5) can't run bf16 in vllm; size vllm for the 16GB
|
|
# card and give the cold start (image pull + model download) headroom.
|
|
VLLM_DTYPE: float16
|
|
VLLM_GPU_TYPE: t4
|
|
SURYA_INFERENCE_STARTUP_TIMEOUT: "1200"
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
- name: Install uv
|
|
uses: astral-sh/setup-uv@v3
|
|
with:
|
|
version: latest
|
|
enable-cache: true
|
|
- name: Set up Python 3.11
|
|
run: uv python install 3.11
|
|
- name: Install dependencies
|
|
run: uv sync --frozen --group dev
|
|
- name: Download benchmark data
|
|
run: |
|
|
wget -O benchmark_data.zip "https://drive.google.com/uc?export=download&id=1NHrdYatR1rtqs2gPVfdvO0BAvocH8CJi"
|
|
unzip -o benchmark_data.zip
|
|
- name: Test detection
|
|
run: uv run surya_detect benchmark_data/pdfs/switch_trans.pdf --page_range 0
|
|
# Spawn the vllm server once and reuse it across the OCR/layout/table
|
|
# steps (--keep_server) instead of paying a cold start three times.
|
|
- name: Test OCR
|
|
run: uv run surya_ocr benchmark_data/pdfs/switch_trans.pdf --page_range 0 --keep_server
|
|
- name: Test layout
|
|
run: uv run surya_layout benchmark_data/pdfs/switch_trans.pdf --page_range 0 --keep_server
|
|
- name: Test table
|
|
run: uv run surya_table benchmark_data/pdfs/switch_trans.pdf --page_range 0 --keep_server
|
|
- name: Test detection folder
|
|
run: uv run surya_detect benchmark_data/pdfs --page_range 0
|