diff --git a/constraints-dev.txt b/constraints-dev.txt index ecf08c90..26dce39e 100644 --- a/constraints-dev.txt +++ b/constraints-dev.txt @@ -83,7 +83,7 @@ jupyterlab-widgets==3.0.15 # via ipywidgets kiwisolver==1.4.9 # via matplotlib lark==1.2.2 # via rfc3987-syntax liger-kernel==0.6.2 # via -r requirements-cuda.txt -llvmlite==0.44.0 # via numba +llvmlite==0.45.0 # via numba markdown==3.8.2 # via tensorboard markdown-it-py==4.0.0 # via rich markupsafe==3.0.2 # via jinja2, nbconvert, werkzeug @@ -107,8 +107,8 @@ ninja==1.13.0 # via deepspeed nodeenv==1.9.1 # via pre-commit notebook==7.4.5 # 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 +numba==0.62.1 # via -r requirements.txt +numpy==2.3.3 # 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.txt b/requirements.txt index e118ea48..721dd914 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,8 +6,8 @@ torch>=2.6.0 transformers>=4.55.0 datasets>=2.15.0 -numba -numpy>=1.26.4,<2.3 +numba>=0.62.0 +numpy>=1.26.4 rich trl>=0.9.4 peft