From ed2da9bf7b9f93ef674b02b807e3192103399563 Mon Sep 17 00:00:00 2001 From: Mathieu Broillet Date: Tue, 20 Aug 2024 22:55:42 +0200 Subject: [PATCH] add useful nodes to comfy and add bnb to forge --- comfyui-rocm/install.sh | 12 +++++++++++- stablediffusion-forge-rocm/install.sh | 3 +++ 2 files changed, 14 insertions(+), 1 deletion(-) diff --git a/comfyui-rocm/install.sh b/comfyui-rocm/install.sh index 885cfa0..ec09f0a 100755 --- a/comfyui-rocm/install.sh +++ b/comfyui-rocm/install.sh @@ -21,10 +21,20 @@ install() { echo "Running $NAME setup..." $python_exec -m pip install -r webui/requirements.txt - # Install manager + cd webui/custom_nodes + + # Install manager git clone https://github.com/ltdrdata/ComfyUI-Manager.git + # Add GGUF support + git clone https://github.com/city96/ComfyUI-GGUF + $python_exec -m pip install --upgrade gguf + + # Add NF4 support + git clone https://github.com/comfyanonymous/ComfyUI_bitsandbytes_NF4.git + $python_exec -m pip install --upgrade ../../../bitsandbytes-rocm-build/bitsandbytes-0.43.3.dev0-cp310-cp310-linux_x86_64.whl # install bitsandbytes for rocm until it is available on pypi + ln -s webui/models models fi diff --git a/stablediffusion-forge-rocm/install.sh b/stablediffusion-forge-rocm/install.sh index bd764b9..da0eb01 100755 --- a/stablediffusion-forge-rocm/install.sh +++ b/stablediffusion-forge-rocm/install.sh @@ -18,6 +18,9 @@ install_stablediffusionforge() { echo "Running StableDiffusionForge setup..." $python_exec webui/launch.py --skip-torch-cuda-test --exit + echo "Adding Flux NF4 support for ROCM" + $python_exec -m pip install --upgrade ../bitsandbytes-rocm-build/bitsandbytes-0.43.3.dev0-cp310-cp310-linux_x86_64.whl # install bitsandbytes for rocm until it is available on pypi + ln -s webui/models models ln -s webui/outputs outputs fi