Skip to content

Commit

Permalink
Fix packages link
Browse files Browse the repository at this point in the history
  • Loading branch information
robballantyne committed Jun 20, 2024
1 parent fdc33b7 commit 6f44421
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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`

Expand Down
4 changes: 2 additions & 2 deletions build/COPY_ROOT_1/opt/ai-dock/bin/build/layer1/amd.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 6f44421

Please sign in to comment.