Merge pull request #260 from sapiosaturn/main

Updating dependency versions in pyproject.toml and README.
huggingface · Dec 5, 2024 · 2cde8f6 · 2cde8f6
2 parents fdd5151 + ed0a0b5
commit 2cde8f6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -34,7 +34,7 @@ Nanotron is a library for pretraining transformer models. It provides a simple a
 ## Installation
 
 ```bash
-# Requirements: Python>=3.10
+# Requirements: Python>=3.10,<3.12
 git clone https://github.com/huggingface/nanotron
 cd nanotron
 pip install --upgrade pip

diff --git a/pyproject.toml b/pyproject.toml
@@ -16,7 +16,7 @@ classifiers = [
 dependencies = [
     "torch>=1.13.1",
     "pyyaml",
-    "numpy",
+    "numpy<2",
     "packaging",
     "safetensors",
     "dacite",
@@ -45,7 +45,7 @@ test = [
 ]
 
 fast-modeling = [
-    "flash-attn>=2.5.0",
+    "flash-attn>=2.5.0,<2.7.0",
 ]
 
 nanosets = [