소스 검색

build(deps): add accelerate package and update requirements

- Add accelerate package to support model training acceleration
- Update requirements.txt to include new dependency
myhloli 7 달 전
부모
커밋
09632dddc1
1개의 변경된 파일1개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      requirements.txt

+ 1 - 0
requirements.txt

@@ -12,4 +12,5 @@ torchvision
 transformers>=4.49.0,<5.0.0
 pdfminer.six==20231228
 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.