deps: cap transformers at v4.44 (#349)

Signed-off-by: Anh Uong <[email protected]>
foundation-model-stack · Sep 25, 2024 · c0c4355 · c0c4355
1 parent 926fb9b
commit c0c4355
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -28,7 +28,7 @@ classifiers=[
 dependencies = [
 "numpy>=1.26.4,<2.0",
 "accelerate>=0.20.3,<0.34",
-"transformers>4.41,<5.0",
+"transformers>4.41,<4.45",
 "torch>=2.2.0,<3.0",
 "sentencepiece>=0.1.99,<0.3",
 "tokenizers>=0.13.3,<1.0",