Add support for ComfyUI

README.md

@@ -4,7 +4,7 @@ Run Stable Diffusion on your Radeon machine with a nice UI without any hassle!
 
 NOTE:
 - Only ROCm supported for now
-- Only AUTOMATIC1111 is confirmed to work for now
+- Supports AUTOMATIC1111 and ComfyUI
 - CUDA support to be worked on later
 - Requires podman-compose 1.1.0 or newer
 
@@ -34,6 +34,8 @@ $ podman generate systemd --new webui-docker_auto_1 > ~/.config/systemd/user/aut
 $ systemctl --user enable --now automatic1111.service
 ```
 
+**You can change `--profile auto` to `--profile comfy` to change AUTOMATIC1111 to ComfyUI**
+
 ## Features
 
 This repository provides multiple UIs for you to play around with stable diffusion:

docker-compose.yml

@@ -59,15 +59,15 @@ services:
   #   environment:
   #     - CLI_ARGS=--no-half --precision full --allow-code --enable-insecure-extension-access --api
   #
-  invoke: &invoke
-    <<: *base_service
-    profiles: ["invoke"]
-    build: ./services/invoke/
-    image: sd-invoke:30
-    environment:
-      - PRELOAD=true
-      # - CLI_ARGS=--xformers
-      - CLI_ARGS=
+  # invoke: &invoke
+  #   <<: *base_service
+  #   profiles: ["invoke"]
+  #   build: ./services/invoke/
+  #   image: sd-invoke:30
+  #   environment:
+  #     - PRELOAD=true
+  #     # - CLI_ARGS=--xformers
+  #     - CLI_ARGS=
 
   # invoke-cpu:
   #   <<: *invoke

services/comfy/Dockerfile

@@ -3,14 +3,14 @@ FROM rocm/pytorch:rocm6.1_ubuntu22.04_py3.10_pytorch_2.1.2
 
 ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
 
-RUN apt-get update && apt-get install -y git && apt-get clean
+RUN apt-get update && apt-get install -y git cargo nasm && apt-get clean
 
 ENV ROOT=/stable-diffusion
 RUN --mount=type=cache,target=/root/.cache/pip,Z \
-  git clone https://github.com/comfyanonymous/ComfyUI.git ${ROOT} && \
+  git clone --recursive https://github.com/comfyanonymous/ComfyUI.git ${ROOT} && \
   cd ${ROOT} && \
   git checkout master && \
-  git reset --hard d1f3637a5a944d0607b899babd8ff11d87100503 && \
+#   git reset --hard d1f3637a5a944d0607b899babd8ff11d87100503 && \
   pip install -r requirements.txt
 
 WORKDIR ${ROOT}