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

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

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