mathieu/ai-suite-rocm-local
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

add useful nodes to comfy and add bnb to forge

Browse Source
This commit is contained in:
Mathieu Broillet 2024-08-20 22:55:42 +02:00
parent c5ba1d3e65
commit ed2da9bf7b
Signed by: mathieu
GPG Key ID: A08E484FE95074C1
2 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
comfyui-rocm/install.sh
View File

@ -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

3
stablediffusion-forge-rocm/install.sh
View File

@ -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