瀏覽代碼

Merge pull request #2114 from myhloli/dev

build(deps): add accelerate package and update requirements https://github.com/opendatalab/MinerU/issues/2112
Xiaomeng Zhao 7 月之前
父節點
當前提交
e3261b0eea
共有 1 個文件被更改,包括 1 次插入0 次删除
  1. 1 0
      requirements.txt

+ 1 - 0
requirements.txt

@@ -12,4 +12,5 @@ torchvision
 transformers>=4.49.0,<5.0.0
 transformers>=4.49.0,<5.0.0
 pdfminer.six==20231228
 pdfminer.six==20231228
 tqdm>=4.67.1
 tqdm>=4.67.1
+accelerate>=1.6.0,<2
 # The requirements.txt must ensure that only necessary external dependencies are introduced. If there are new dependencies to add, please contact the project administrator.
 # The requirements.txt must ensure that only necessary external dependencies are introduced. If there are new dependencies to add, please contact the project administrator.