Merge pull request #2062 from bmaltais/dev

23.0.4
bmaltais · Mar 11, 2024 · 0eaf835 · 0eaf835
2 parents d9e2e36 + eb9d7bc
commit 0eaf835
Show file tree

Hide file tree

Showing 11 changed files with 14 additions and 6 deletions.
diff --git a/.release b/.release
@@ -1 +1 @@
-v23.0.3
+v23.0.4
diff --git a/README.md b/README.md
@@ -37,6 +37,7 @@ The GUI allows you to set the training parameters and generate and run the requi
     - [No module called tkinter](#no-module-called-tkinter)
   - [SDXL training](#sdxl-training)
   - [Change History](#change-history)
+    - [2024/03/10 (v23.0.4)](#20240310-v2304)
     - [2024/03/10 (v23.0.3)](#20240310-v2303)
     - [2024/03/10 (v23.0.2)](#20240310-v2302)
     - [2024/03/09 (v23.0.1)](#20240309-v2301)
@@ -362,6 +363,11 @@ The documentation in this section will be moved to a separate document later.
 
 ## Change History
 
+### 2024/03/10 (v23.0.4)
+
+- Update bitsandbytes to 0.43.0
+- Add packaging to runpod setup
+
 ### 2024/03/10 (v23.0.3)
 
 - Fix bug with setup

diff --git a/requirements_linux.txt b/requirements_linux.txt
@@ -1,4 +1,4 @@
 torch==2.1.2+cu118 torchvision==0.16.2+cu118 xformers==0.0.23.post1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
-bitsandbytes==0.41.2
+bitsandbytes==0.43.0
 tensorboard==2.15.2 tensorflow==2.15.0.post1
 -r requirements.txt
diff --git a/requirements_linux_docker.txt b/requirements_linux_docker.txt
@@ -1,5 +1,5 @@
 xformers>=0.0.20
-bitsandbytes==0.41.1
+bitsandbytes==0.43.0
 accelerate==0.25.0
 tensorboard==2.15.2
 tensorflow==2.15.0.post1
diff --git a/requirements_runpod.txt b/requirements_runpod.txt
@@ -1,5 +1,5 @@
 torch==2.1.2+cu118 torchvision==0.16.2+cu118 xformers==0.0.23.post1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 # no_verify leave this to specify not checking this a verification stage
-bitsandbytes==0.41.2
+bitsandbytes==0.43.0
 tensorboard==2.15.2 tensorflow==2.15.0.post1 wheel
 tensorrt
 -r requirements.txt
diff --git a/requirements_windows_torch2.txt b/requirements_windows_torch2.txt
@@ -1,7 +1,5 @@
 torch==2.1.2+cu118 torchvision==0.16.2+cu118 torchaudio==2.1.2+cu118 xformers==0.0.23.post1+cu118 --index-url https://download.pytorch.org/whl/cu118
 nvidia-cudnn-cu11==8.9.5.29
-# https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.41.2.post2-py3-none-win_amd64.whl # no_verify
-# https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.41.1-py3-none-win_amd64.whl # no_verify
 bitsandbytes==0.43.0
 tensorboard==2.14.1 tensorflow==2.14.0
 -r requirements.txt
diff --git a/setup-runpod.sh b/setup-runpod.sh
@@ -21,6 +21,10 @@ fi
 echo "Activating venv..."
 source "$SCRIPT_DIR/venv/bin/activate" || exit 1
 
+# Install packaging
+echo "Installing the python packaging module..."
+pip install packaging
+
 # Run setup_linux.py script with platform requirements
 echo "Running setup_linux.py..."
 python "$SCRIPT_DIR/setup/setup_linux.py" --platform-requirements-file=requirements_runpod.txt --show_stdout --no_run_accelerate

diff --git a/test/ft/.keep b/test/ft/.keep
diff --git a/test/log/.keep b/test/log/.keep
diff --git a/test/logs/.keep b/test/logs/.keep
diff --git a/test/output/.keep b/test/output/.keep