Skip to content

Commit c4dbb91

Browse files
committed
chore: update dependency versions and add addict to local requirements and engine checks
1 parent 2104d66 commit c4dbb91

3 files changed

Lines changed: 7 additions & 2 deletions

File tree

ocr_pipeline/services/local_ocr_engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ async def _ensure_loaded(self) -> None:
122122
def _load_blocking(self) -> None:
123123
missing = [
124124
package
125-
for package in ("torch", "transformers", "easydict")
125+
for package in ("torch", "transformers", "tokenizers", "addict", "easydict")
126126
if find_spec(package) is None
127127
]
128128
if missing:

requirements-local.txt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,10 @@
66
# CUDA-matched wheels from https://pytorch.org/ instead of letting pip pick.
77
torch>=2.4.0
88
torchvision>=0.19.0
9-
transformers>=4.46.0
9+
transformers==4.46.3
10+
tokenizers==0.20.3
1011
accelerate>=0.34.0
1112
einops>=0.8.0
1213
sentencepiece>=0.2.0
14+
addict>=2.4.0
1315
easydict>=1.13

tests/test_requirements.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,4 +5,7 @@ def test_local_requirements_include_deepseek_remote_code_dependencies():
55
requirements = (Path(__file__).parents[1] / "requirements-local.txt").read_text(
66
encoding="utf-8"
77
)
8+
assert "transformers==4.46.3" in requirements
9+
assert "tokenizers==0.20.3" in requirements
10+
assert "addict" in requirements
811
assert "easydict" in requirements

0 commit comments

Comments
 (0)