add useful nodes to comfy and add bnb to forge

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
 

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