Add AQLM support (transformers loader) (#5466)

2024-09-19 15:56:56 +00:00 · 2024-03-08 17:30:36 -03:00 · 2024-03-08 17:30:36 -03:00 · 0e6eb7c27a
commit 0e6eb7c27a
parent 2681f6f640
9 changed files with 9 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,4 +1,5 @@
 accelerate==0.27.*
+aqlm[gpu,cpu]==1.1.0
 colorama
 datasets
 einops
--- a/requirements_amd.txt
+++ b/requirements_amd.txt
@ -1,4 +1,5 @@
 accelerate==0.27.*
+aqlm[gpu,cpu]==1.1.0
 colorama
 datasets
 einops
--- 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
--- 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
--- 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
--- 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
--- 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
--- a/requirements_noavx2.txt
+++ b/requirements_noavx2.txt
@ -1,4 +1,5 @@
 accelerate==0.27.*
+aqlm[gpu,cpu]==1.1.0
 colorama
 datasets
 einops
--- a/requirements_nowheels.txt
+++ b/requirements_nowheels.txt
@ -1,4 +1,5 @@
 accelerate==0.27.*
+aqlm[gpu,cpu]==1.1.0
 colorama
 datasets
 einops