diff --git a/requirements.txt b/requirements.txt index 1041025c..54bd414a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ accelerate==0.33.* aqlm[gpu,cpu]==1.1.6; platform_system == "Linux" auto-gptq==0.7.1 -bitsandbytes==0.43.* +bitsandbytes==0.44.* colorama datasets einops diff --git a/requirements_noavx2.txt b/requirements_noavx2.txt index 69d880ac..39bfb0bd 100644 --- a/requirements_noavx2.txt +++ b/requirements_noavx2.txt @@ -1,7 +1,7 @@ accelerate==0.33.* aqlm[gpu,cpu]==1.1.6; platform_system == "Linux" auto-gptq==0.7.1 -bitsandbytes==0.43.* +bitsandbytes==0.44.* colorama datasets einops