requirements: Adjust numpy dependency range

According to comments on #61, numpy 2 doesn't work with this library right now. Instead of pinning to a specific 1.x version, allow a range up to, but not including 2.0.0. The specifics on the lower end come from what's used in `instructlab/instructlab`. Signed-off-by: Russell Bryant <[email protected]>
instructlab · Jun 25, 2024 · 12e8ff5 · 12e8ff5
1 parent 700dad0
commit 12e8ff5
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -7,7 +7,13 @@ torch>=2.3.0a0
 transformers>=4.41.2
 datasets>=2.15.0
 numba
-numpy==1.26.4
+# Note: numpy ranges copied from instructlab/instructlab
+#
+# HabanaLabs / Intel Gaudi env comes with Python 3.10 and slightly older
+# versions of some dependencies. Use '3.10' as an indicator.
+# Habana installer has NumPy 1.23.5
+numpy>=1.23.5,<2.0.0 ; python_version == '3.10'
+numpy>=1.26.4,<2.0.0 ; python_version != '3.10'
 rich
 instructlab-dolomite
 trl>=0.9.4
@@ -16,4 +22,4 @@ pydantic>=2.7.0
 
 # deepspeed needs to be at the bottom or it'll break during installation
 deepspeed>=0.14.3
-aiofiles>=23.2.1
+aiofiles>=23.2.1