From caa22aeb6d87f1c2ec0b49a9656f13c842b6a5a5 Mon Sep 17 00:00:00 2001 From: bmaltais Date: Mon, 26 Jun 2023 15:13:51 -0400 Subject: [PATCH] Revving back bitsandbytes for linux causing training issues --- requirements_linux.txt | 2 +- requirements_linux_docker.txt | 2 +- requirements_runpod.txt | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/requirements_linux.txt b/requirements_linux.txt index e69632f..a609ce5 100644 --- a/requirements_linux.txt +++ b/requirements_linux.txt @@ -1,4 +1,4 @@ torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 # no_verify leave this to specify not checking this a verification stage -xformers==0.0.20 bitsandbytes==0.39.1 +xformers==0.0.20 bitsandbytes==0.35.0 accelerate==0.15.0 tensorboard==2.12.1 tensorflow==2.12.0 -r requirements.txt diff --git a/requirements_linux_docker.txt b/requirements_linux_docker.txt index 03d9d30..03b6bba 100644 --- a/requirements_linux_docker.txt +++ b/requirements_linux_docker.txt @@ -1,5 +1,5 @@ xformers==0.0.20 -bitsandbytes==0.39.1 +bitsandbytes==0.35.0 accelerate==0.15.0 tensorboard==2.12.1 tensorflow==2.12.0 diff --git a/requirements_runpod.txt b/requirements_runpod.txt index e69632f..f1a4870 100644 --- a/requirements_runpod.txt +++ b/requirements_runpod.txt @@ -1,4 +1,4 @@ -torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 # no_verify leave this to specify not checking this a verification stage -xformers==0.0.20 bitsandbytes==0.39.1 +--no-cache-dir torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 # no_verify leave this to specify not checking this a verification stage +xformers==0.0.20 bitsandbytes==0.35.0 accelerate==0.15.0 tensorboard==2.12.1 tensorflow==2.12.0 -r requirements.txt