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

pull/142/head
Robert Sachunsky 3 weeks ago
parent 871d7bfc5a
commit f765e2603b

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

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

Loading…
Cancel
Save