소스 검색

Require accelerate>=0.20.3 as transformers do (#383)

Alexander Borzunov 2 년 전
부모
커밋
e9a20e7e53
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      setup.cfg

+ 1 - 1
setup.cfg

@@ -33,7 +33,7 @@ python_requires = >=3.8,<3.11
 install_requires =
     torch>=1.12
     bitsandbytes==0.40.1.post1
-    accelerate>=0.16.0,<0.21.0
+    accelerate>=0.20.3,<0.21.0
     huggingface-hub>=0.11.1,<1.0.0
     tokenizers>=0.13.3
     transformers>=4.31.0,<5.0.0