yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-10-26 23:53:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pyproject dependency cleanup

Browse Source
This commit is contained in:
Jake Poznanski 2024-09-20 08:22:10 -07:00
parent 962fb7eb6d
commit dc86a99a97
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pyproject.toml
View File

@ -20,6 +20,7 @@ authors = [
requires-python = ">=3.8"
dependencies = [
"cached-path",
"smart_open",
"pypdf",
"pymupdf",
"pypdfium2",
@ -52,13 +53,19 @@ dev = [
"sphinx-copybutton==0.5.2",
"sphinx-autobuild==2021.3.14",
"sphinx-autodoc-typehints==1.23.3",
"packaging"
"packaging",
"necessary",
]
[project.optional-dependencies]
train = [
"torch",
"torchvision",
"accelerate",
"datasets",
"peft",
"wandb",
"omegaconf",
"s3fs",
"transformers @ git+https://github.com/huggingface/transformers.git@f9b44097261270377199bb0fcbceb01fd9f7d0c0"
]