From 0e6eb7c27a877c3b18e7125995e5bc47b7f74767 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Fri, 8 Mar 2024 17:30:36 -0300 Subject: [PATCH] Add AQLM support (transformers loader) (#5466) --- requirements.txt | 1 + requirements_amd.txt | 1 + requirements_amd_noavx2.txt | 1 + requirements_apple_intel.txt | 1 + requirements_apple_silicon.txt | 1 + requirements_cpu_only.txt | 1 + requirements_cpu_only_noavx2.txt | 1 + requirements_noavx2.txt | 1 + requirements_nowheels.txt | 1 + 9 files changed, 9 insertions(+) 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