yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-06-27 04:00:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Even newer mypy crashes still

Browse Source
This commit is contained in:
Jake Poznanski 2025-01-30 14:32:08 -08:00
parent c74d47a553
commit 10094ffc19
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

@ -51,7 +51,7 @@ Changelog = "https://github.com/allenai/olmocr/blob/main/CHANGELOG.md"
[project.optional-dependencies]
dev = [
"ruff",
"mypy>=1.0,<1.5",
"mypy",
"black",
"isort",
"pytest",

3
tests/test_sglang.py
View File

@ -24,7 +24,6 @@ from transformers import AutoProcessor, AutoTokenizer, Qwen2VLForConditionalGene
from olmocr.pipeline import (
SGLANG_SERVER_PORT,
build_page_query,
download_directory,
get_anchor_text,
render_pdf_to_base64png,
sglang_server_ready,
@ -32,6 +31,8 @@ from olmocr.pipeline import (
)
from olmocr.prompts import PageResponse
MODEL_FINETUNED_PATH = (
"s3://ai2-oe-data/jakep/experiments/qwen2vl-pdf/v1/models/jakep/Qwen_Qwen2-VL-7B-Instruct-e4ecf8-01JAH8GMWHTJ376S2N7ETXRXH4/checkpoint-9500/bf16/"
)