diff --git a/setup.py b/setup.py
index 53a0b84..6362945 100644
--- a/setup.py
+++ b/setup.py
@@ -57,7 +57,7 @@ def read(file_name):
 
     python_requires='>=3.8',
     install_requires=[
-        'numpy==1.26.4',
+        'numpy==2.1.2',
         'tensorflow==2.11.1',
         'tqdm==4.66.5',
         'Pillow==10.4.0',