Set ranges for deps

Signed-off-by: Thara Palanivel <[email protected]>
foundation-model-stack · Jan 15, 2025 · d8cd63e · d8cd63e
1 parent 0eaf455
commit d8cd63e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -21,12 +21,12 @@ classifiers=[
 ]
 dynamic = ["version"]
 dependencies = [
-"numpy>=1.26.4,<2.0",
+"numpy>=2.2.0",
 "accelerate>=0.20.3,!=0.34,<1.1",
-"transformers>=4.45,<4.46",
+"transformers>=4.45,<4.48",
 "torch>=2.2.0,<2.4", 
 "tqdm>=4.66.2,<5.0",
-"datasets>=2.15.0,<3.0",
+"datasets>=3.0.0,<4.0",
 "ninja>=1.11.1.1,<2.0",
 "tensorboard",
 "notebook",