diff --git a/requirements.txt b/requirements.txt
index 4fef7b6..e55fd81 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 torch>=2.3.0
 timm>=1.0.3
-numpy==1.26.4
+numpy<=2.0
 huggingface_hub>=0.23.1
 einops>=0.8.0
 transformers>=4.41.1