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