Add AQLM support (transformers loader) (#5466)

oobabooga · Mar 8, 2024 · 0e6eb7c · 0e6eb7c
1 parent 2681f6f
commit 0e6eb7c
Show file tree

Hide file tree

Showing 9 changed files with 9 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -1,4 +1,5 @@
 accelerate==0.27.*
+aqlm[gpu,cpu]==1.1.0
 colorama
 datasets
 einops