diff --git a/requirements.txt b/requirements.txt index 3028fe7a..ed4941d4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ accelerate==0.27.* +aqlm[gpu,cpu]==1.1.0 colorama datasets einops diff --git a/requirements_amd.txt b/requirements_amd.txt index f43f6ac4..b0d7dc75 100644 --- a/requirements_amd.txt +++ b/requirements_amd.txt @@ -1,4 +1,5 @@ accelerate==0.27.* +aqlm[gpu,cpu]==1.1.0 colorama datasets einops diff --git a/requirements_amd_noavx2.txt b/requirements_amd_noavx2.txt index 37dfcea2..f8c53550 100644 --- a/requirements_amd_noavx2.txt +++ b/requirements_amd_noavx2.txt @@ -1,4 +1,5 @@ accelerate==0.27.* +aqlm[gpu,cpu]==1.1.0 colorama datasets einops diff --git a/requirements_apple_intel.txt b/requirements_apple_intel.txt index 88d578ad..cb2254af 100644 --- a/requirements_apple_intel.txt +++ b/requirements_apple_intel.txt @@ -1,4 +1,5 @@ accelerate==0.27.* +aqlm[gpu,cpu]==1.1.0 colorama datasets einops diff --git a/requirements_apple_silicon.txt b/requirements_apple_silicon.txt index 13767480..2ad17121 100644 --- a/requirements_apple_silicon.txt +++ b/requirements_apple_silicon.txt @@ -1,4 +1,5 @@ accelerate==0.27.* +aqlm[gpu,cpu]==1.1.0 colorama datasets einops diff --git a/requirements_cpu_only.txt b/requirements_cpu_only.txt index 92b5b969..25bf2292 100644 --- a/requirements_cpu_only.txt +++ b/requirements_cpu_only.txt @@ -1,4 +1,5 @@ accelerate==0.27.* +aqlm[gpu,cpu]==1.1.0 colorama datasets einops diff --git a/requirements_cpu_only_noavx2.txt b/requirements_cpu_only_noavx2.txt index 01964fde..02bbf2c0 100644 --- a/requirements_cpu_only_noavx2.txt +++ b/requirements_cpu_only_noavx2.txt @@ -1,4 +1,5 @@ accelerate==0.27.* +aqlm[gpu,cpu]==1.1.0 colorama datasets einops diff --git a/requirements_noavx2.txt b/requirements_noavx2.txt index a4f9bcb9..c81a5472 100644 --- a/requirements_noavx2.txt +++ b/requirements_noavx2.txt @@ -1,4 +1,5 @@ accelerate==0.27.* +aqlm[gpu,cpu]==1.1.0 colorama datasets einops diff --git a/requirements_nowheels.txt b/requirements_nowheels.txt index 7166015d..a7179315 100644 --- a/requirements_nowheels.txt +++ b/requirements_nowheels.txt @@ -1,4 +1,5 @@ accelerate==0.27.* +aqlm[gpu,cpu]==1.1.0 colorama datasets einops