diff --git a/constraints-dev.txt b/constraints-dev.txt index 28150b85..797e76b9 100644 --- a/constraints-dev.txt +++ b/constraints-dev.txt @@ -109,7 +109,7 @@ nodeenv==1.9.1 # via pre-commit notebook==7.4.3 # via jupyter notebook-shim==0.2.4 # via jupyterlab, notebook numba==0.61.2 # via -r requirements.txt -numpy==1.26.4 # via accelerate, bitsandbytes, contourpy, datasets, deepspeed, matplotlib, numba, pandas, peft, tensorboard, transformers, -r requirements-dev.txt, -r requirements.txt +numpy==2.2.6 # via accelerate, bitsandbytes, contourpy, datasets, deepspeed, matplotlib, numba, pandas, peft, tensorboard, transformers, -r requirements-dev.txt, -r requirements.txt nvidia-cublas-cu12==12.4.5.8 # via nvidia-cudnn-cu12, nvidia-cusolver-cu12, torch nvidia-cuda-cupti-cu12==12.4.127 # via torch nvidia-cuda-nvrtc-cu12==12.4.127 # via torch diff --git a/requirements-dev.txt b/requirements-dev.txt index 3fb79e43..7b106f2e 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -4,7 +4,6 @@ isort matplotlib -numpy pre-commit>=3.0.4 pylint>=2.16.2 pylint-pydantic diff --git a/requirements.txt b/requirements.txt index 4396521c..c4ca2f64 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,7 @@ transformers>=4.45.2 datasets>=2.15.0 numba -numpy>=1.26.4,<2.0.0 +numpy>=1.26.4 rich instructlab-dolomite>=0.2.0 trl>=0.9.4