瀏覽代碼

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 年之前
父節點
當前提交
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.