Explorar o código

feat(requirements): add torch and transformers libraries

Introduce torch and transformers libraries to support new ML features.Ensure version compatibility by adding torch version within the range 2.2.2 to 2.3.1and include the necessary transformers library.
myhloli hai 1 ano
pai
achega
656154558a
Modificáronse 1 ficheiros con 2 adicións e 0 borrados
  1. 2 0
      requirements.txt

+ 2 - 0
requirements.txt

@@ -9,4 +9,6 @@ pydantic>=2.7.2,<2.8.0
 PyMuPDF>=1.24.9
 scikit-learn>=1.0.2
 wordninja>=2.0.0
+torch>=2.2.2,<=2.3.1
+transformers
 # The requirements.txt must ensure that only necessary external dependencies are introduced. If there are new dependencies to add, please contact the project administrator.