qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-07-23 09:49:58 +02:00
Code Issues Projects Releases Wiki Activity

move Torch to optional dependencies (to avoid clash with TF over CuDNN)

Browse source
This commit is contained in:
Robert Sachunsky 2024-12-04 15:57:13 +00:00
parent 871d7bfc5a
commit f765e2603b
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
pyproject.toml
View file

@ -25,6 +25,10 @@ classifiers = [
"Topic :: Scientific/Engineering :: Image Processing",
]
[project.optional-dependencies]
OCR = ["torch <= 2.0.1", "transformers <= 4.30.2"]
plotting = ["matplotlib"]
[project.scripts]
eynollah = "eynollah.cli:main"
ocrd-eynollah-segment = "eynollah.ocrd_cli:main"

4
requirements.txt
View file

@ -4,8 +4,4 @@ numpy <1.24.0
scikit-learn >= 0.23.2
tensorflow < 2.13
imutils >= 0.5.3
matplotlib
setuptools >= 50
transformers <= 4.30.2
torch <= 2.0.1
numba <= 0.58.1