Sfoglia il codice sorgente

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

Alexander Borzunov 2 anni fa
parent
commit
e9a20e7e53
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      setup.cfg

+ 1 - 1
setup.cfg

@@ -33,7 +33,7 @@ python_requires = >=3.8,<3.11
 install_requires =
 install_requires =
     torch>=1.12
     torch>=1.12
     bitsandbytes==0.40.1.post1
     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
     huggingface-hub>=0.11.1,<1.0.0
     tokenizers>=0.13.3
     tokenizers>=0.13.3
     transformers>=4.31.0,<5.0.0
     transformers>=4.31.0,<5.0.0