update accelerate version

gaussalgo · Dec 14, 2023 · 9257095 · 9257095
1 parent 028b7db
commit 9257095
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -32,6 +32,7 @@
         "torch>=1.7",
         "transformers>=4.33.3",
         "sentencepiece",
+        "accelerate>=0.20.1"
     ],
     test_require=[
         "pytest"