Merge branch 'main' into jim_alex_dev_image

merging updates from main
foundation-model-stack · Oct 18, 2024 · 08739d1 · 08739d1
2 parents 880e02c + d58faa5
commit 08739d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -29,7 +29,7 @@ dependencies = [
 "numpy>=1.26.4,<2.0",
 "accelerate>=0.20.3,!=0.34,<1.1",
 "transformers>4.41,<4.50",
-"torch>=2.2.0,<3.0",
+"torch>=2.2.0,<2.5",
 "sentencepiece>=0.1.99,<0.3",
 "tokenizers>=0.13.3,<1.0",
 "tqdm>=4.66.2,<5.0",