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

specify gpu deps in pyproject

Browse Source
This commit is contained in:
Chris Wilhelm 2025-03-13 13:26:04 -07:00
parent 29b9054749
commit f1524957b1
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pyproject.toml
View File

@ -48,6 +48,11 @@ Changelog = "https://github.com/allenai/olmocr/blob/main/CHANGELOG.md"
# Documentation = "https://olmocr.readthedocs.io/" # Documentation = "https://olmocr.readthedocs.io/"
[project.optional-dependencies] [project.optional-dependencies]
gpu = [
"sgl-kernel==0.0.3.post1",
"sglang[all]==0.4.2",
"flashinfer"
]
dev = [ dev = [
"ruff", "ruff",
"mypy", "mypy",
@ -120,6 +125,9 @@ exclude = [
[tool.setuptools] [tool.setuptools]
include-package-data = true include-package-data = true
find-links = [
"https://flashinfer.ai/whl/cu124/torch2.4/flashinfer/"
]
[tool.setuptools.package-data] [tool.setuptools.package-data]
olmocr = [ olmocr = [
@ -184,4 +192,4 @@ log_format = "%(asctime)s - %(levelname)s - %(name)s - %(message)s"
log_level = "DEBUG" log_level = "DEBUG"
markers = [ markers = [
"nonci: mark test as not intended for CI runs" "nonci: mark test as not intended for CI runs"
] ]