From 1ad583363fdab8f2db2b8c47080af42922d7da2b Mon Sep 17 00:00:00 2001 From: Oleg Silkin <97077423+RobotSail@users.noreply.github.com> Date: Fri, 10 Oct 2025 09:18:55 -0400 Subject: [PATCH 1/5] removes version cap on numpy --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index e118ea48..a502dc7e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,7 @@ transformers>=4.55.0 datasets>=2.15.0 numba -numpy>=1.26.4,<2.3 +numpy>=1.26.4 rich trl>=0.9.4 peft From e79091b7c11e1293eacbf7c66bf7b89a0318e481 Mon Sep 17 00:00:00 2001 From: Oleg Silkin <97077423+RobotSail@users.noreply.github.com> Date: Fri, 10 Oct 2025 12:33:32 -0400 Subject: [PATCH 2/5] set the minimum version of numba to the first release which effectively uncaps numpy Signed-off-by: Oleg Silkin <97077423+RobotSail@users.noreply.github.com> --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index a502dc7e..b230316f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,7 +6,7 @@ torch>=2.6.0 transformers>=4.55.0 datasets>=2.15.0 -numba +numba>= 0.62.0 numpy>=1.26.4 rich trl>=0.9.4 From 8857376827faa8d8938b9d38dfc9b2fe347cc64d Mon Sep 17 00:00:00 2001 From: Oleg Silkin <97077423+RobotSail@users.noreply.github.com> Date: Fri, 10 Oct 2025 21:15:37 -0400 Subject: [PATCH 3/5] update constraints too --- constraints-dev.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/constraints-dev.txt b/constraints-dev.txt index ecf08c90..dde2d79c 100644 --- a/constraints-dev.txt +++ b/constraints-dev.txt @@ -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 From 392ec682858f37b5656ed4920f17dddc56f94a17 Mon Sep 17 00:00:00 2001 From: Oleg Silkin <97077423+RobotSail@users.noreply.github.com> Date: Fri, 10 Oct 2025 22:44:19 -0400 Subject: [PATCH 4/5] bump min constraint of llvmlite --- constraints-dev.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/constraints-dev.txt b/constraints-dev.txt index dde2d79c..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 From a8d03b53d1ac8079468ba65e85be407de267489c Mon Sep 17 00:00:00 2001 From: Oleg Silkin <97077423+RobotSail@users.noreply.github.com> Date: Mon, 13 Oct 2025 10:51:30 -0400 Subject: [PATCH 5/5] Update requirements.txt Signed-off-by: Oleg Silkin <97077423+RobotSail@users.noreply.github.com> --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index b230316f..721dd914 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,7 +6,7 @@ torch>=2.6.0 transformers>=4.55.0 datasets>=2.15.0 -numba>= 0.62.0 +numba>=0.62.0 numpy>=1.26.4 rich trl>=0.9.4