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