mirror of
https://github.com/qurator-spk/eynollah.git
synced 2025-07-01 06:59:54 +02:00
move Torch to optional dependencies (to avoid clash with TF over CuDNN)
This commit is contained in:
parent
871d7bfc5a
commit
f765e2603b
2 changed files with 4 additions and 4 deletions
|
@ -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,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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue