From 6f4442156a6dfd71122284bbd6b31d71f196ba56 Mon Sep 17 00:00:00 2001 From: Rob Ballantyne Date: Thu, 20 Jun 2024 21:14:36 +0100 Subject: [PATCH] Fix packages link --- README.md | 2 +- build/COPY_ROOT_1/opt/ai-dock/bin/build/layer1/amd.sh | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index f607ef3..0b268c0 100644 --- a/README.md +++ b/README.md @@ -31,7 +31,7 @@ Tags follow these patterns: - `:latest-rocm` → `:v2-rocm-6.0-core-22.04-v24.1.4` -Browse [here](https://github.com/ai-dock/stable-diffusion-webui/pkgs/container/stable-diffusion-webui) for an image suitable for your target environment. +Browse [here](https://github.com/ai-dock/kohya_ss/pkgs/container/kohya_ss) for an image suitable for your target environment. Supported Python versions: `3.10` diff --git a/build/COPY_ROOT_1/opt/ai-dock/bin/build/layer1/amd.sh b/build/COPY_ROOT_1/opt/ai-dock/bin/build/layer1/amd.sh index 2d0930b..8ce1daf 100755 --- a/build/COPY_ROOT_1/opt/ai-dock/bin/build/layer1/amd.sh +++ b/build/COPY_ROOT_1/opt/ai-dock/bin/build/layer1/amd.sh @@ -17,10 +17,10 @@ build_amd_install_bitsandbytes() { git clone --recurse https://github.com/ROCm/bitsandbytes cd bitsandbytes git checkout rocm_enabled - "$KOHYA_VENV_PIP" install -r requirements-dev.txt + "$KOHYA_VENV_PIP" install --no-cache-dir -r requirements-dev.txt cmake -DCOMPUTE_BACKEND=hip -S . #Use -DBNB_ROCM_ARCH="gfx90a;gfx942" to target specific gpu arch make - "$KOHYA_VENV_PIP" install . + "$KOHYA_VENV_PIP" install --no-cache-dir . cd /tmp rm -rf /tmp/bitsandbytes if [[ $ROCM_LEVEL != "devel" ]]; then