Эх сурвалжийг харах

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 1 жил өмнө
parent
commit
656154558a
1 өөрчлөгдсөн 2 нэмэгдсэн , 0 устгасан
  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.