dep: cap transformers version (#335)

- FSDP bug in accelerate v0.34 Signed-off-by: Anh Uong <[email protected]>
foundation-model-stack · Sep 10, 2024 · 553fa22 · 553fa22
1 parent 32b751c
commit 553fa22
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -27,7 +27,7 @@ classifiers=[
 ]
 dependencies = [
 "numpy>=1.26.4,<2.0",
-"accelerate>=0.20.3,<0.40",
+"accelerate>=0.20.3,<0.34",
 "transformers>4.41,<5.0",
 "torch>=2.2.0,<3.0",
 "sentencepiece>=0.1.99,<0.3",