From 03c191032e720110fa41b04ee58c34e62519b956 Mon Sep 17 00:00:00 2001
From: Varun Singh <v7singh@ucsd.edu>
Date: Wed, 4 Dec 2024 23:11:22 -0700
Subject: [PATCH 1/2] updated dependency versions specified for compatibility

---
 pyproject.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 9794ab78..390c32c4 100644
--- 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 = [

From ed0a0b50dfea74855d2fa57049d591a0d76f0b4b Mon Sep 17 00:00:00 2001
From: Varun Singh <v7singh@ucsd.edu>
Date: Wed, 4 Dec 2024 23:23:32 -0700
Subject: [PATCH 2/2] edited to state python version <3.12

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 6bdc917d..c8837564 100644
--- 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