From b3ade5832b6f79f1521e376aec9a31be0ea4f45f Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Sun, 10 Mar 2024 09:41:17 -0700 Subject: [PATCH] Keep AQLM only for Linux (fails to install on Windows) --- requirements.txt | 2 +- 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 | 2 +- requirements_nowheels.txt | 1 - 9 files changed, 2 insertions(+), 9 deletions(-) diff --git a/requirements.txt b/requirements.txt index f66c0167..d608907a 100644 --- 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 diff --git a/requirements_amd.txt b/requirements_amd.txt index b4cf3f46..57bd8d64 100644 --- a/requirements_amd.txt +++ b/requirements_amd.txt @@ -1,5 +1,4 @@ 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 d2936848..04f6a1a4 100644 --- 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 diff --git a/requirements_apple_intel.txt b/requirements_apple_intel.txt index bf48e023..7c451c80 100644 --- 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 diff --git a/requirements_apple_silicon.txt b/requirements_apple_silicon.txt index 3298dfbd..1178b8b4 100644 --- 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 diff --git a/requirements_cpu_only.txt b/requirements_cpu_only.txt index 0f4a734f..beda6bd8 100644 --- 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 diff --git a/requirements_cpu_only_noavx2.txt b/requirements_cpu_only_noavx2.txt index df470529..3300564e 100644 --- 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 diff --git a/requirements_noavx2.txt b/requirements_noavx2.txt index 5eeaca1d..0c92c8d2 100644 --- 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 diff --git a/requirements_nowheels.txt b/requirements_nowheels.txt index a7179315..7166015d 100644 --- a/requirements_nowheels.txt +++ b/requirements_nowheels.txt @@ -1,5 +1,4 @@ accelerate==0.27.* -aqlm[gpu,cpu]==1.1.0 colorama datasets einops