Keep AQLM only for Linux (fails to install on Windows)

2024-11-22 08:07:56 +01:00 · 2024-03-10 09:41:17 -07:00 · 2024-03-10 09:41:17 -07:00 · b3ade5832b
commit b3ade5832b
parent 67b24b0b88
9 changed files with 2 additions and 9 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,5 +1,5 @@
 accelerate==0.27.*
-aqlm[gpu,cpu]==1.1.0
+aqlm[gpu,cpu]==1.1.0; platform_system == "Linux"
 bitsandbytes==0.43.*
 colorama
 datasets
--- a/requirements_amd.txt
+++ b/requirements_amd.txt
@ -1,5 +1,4 @@
 accelerate==0.27.*
-aqlm[gpu,cpu]==1.1.0
 colorama
 datasets
 einops
--- a/requirements_amd_noavx2.txt
+++ b/requirements_amd_noavx2.txt
@ -1,5 +1,4 @@
 accelerate==0.27.*
-aqlm[gpu,cpu]==1.1.0
 colorama
 datasets
 einops
--- a/requirements_apple_intel.txt
+++ b/requirements_apple_intel.txt
@ -1,5 +1,4 @@
 accelerate==0.27.*
-aqlm[gpu,cpu]==1.1.0
 colorama
 datasets
 einops
--- a/requirements_apple_silicon.txt
+++ b/requirements_apple_silicon.txt
@ -1,5 +1,4 @@
 accelerate==0.27.*
-aqlm[gpu,cpu]==1.1.0
 colorama
 datasets
 einops
--- a/requirements_cpu_only.txt
+++ b/requirements_cpu_only.txt
@ -1,5 +1,4 @@
 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,5 +1,4 @@
 accelerate==0.27.*
-aqlm[gpu,cpu]==1.1.0
 colorama
 datasets
 einops
--- a/requirements_noavx2.txt
+++ b/requirements_noavx2.txt
@ -1,5 +1,5 @@
 accelerate==0.27.*
-aqlm[gpu,cpu]==1.1.0
+aqlm[gpu,cpu]==1.1.0; platform_system == "Linux"
 bitsandbytes==0.43.*
 colorama
 datasets
--- a/requirements_nowheels.txt
+++ b/requirements_nowheels.txt
@ -1,5 +1,4 @@
 accelerate==0.27.*
-aqlm[gpu,cpu]==1.1.0
 colorama
 datasets
 einops