From eea8521f4665d3d4e17a2c9c3545da22dd9d1438 Mon Sep 17 00:00:00 2001 From: Mustafa Eyceoz Date: Thu, 25 Jul 2024 12:13:51 -0400 Subject: [PATCH] Update requirements to have bitsandbytes min and dolomite min (#153) * Update requirements-cuda.txt to have bitsandbytes min Signed-off-by: Mustafa Eyceoz * Update requirements.txt for dolomite min Signed-off-by: Mustafa Eyceoz --------- Signed-off-by: Mustafa Eyceoz --- requirements-cuda.txt | 2 +- requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 01556f33..1bed61eb 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -1,2 +1,2 @@ flash-attn>=2.4.0 -bitsandbytes +bitsandbytes>=0.43.1 diff --git a/requirements.txt b/requirements.txt index 84208b13..442e5e4a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -16,7 +16,7 @@ numba numpy>=1.23.5,<2.0.0 ; python_version == '3.10' numpy>=1.26.4,<2.0.0 ; python_version != '3.10' rich -instructlab-dolomite +instructlab-dolomite>=0.1.1 trl>=0.9.4 peft pydantic>=2.7.0