Update docker

Add path for docker
Add Dockerfile for AMD

docker/docker-compose.yml

@@ -3,7 +3,9 @@ version: "3.9"
 
 x-base_service: &base_service
   image: comfyui:1
-  build: .
+  build:
+    context: docker_files
+    dockerfile: default.Dockerfile
 
   environment:
     - CLI_ARGS=--listen --port 8188
@@ -11,10 +13,11 @@ x-base_service: &base_service
   ports:
     - "8188:8188"
   volumes:
-    - ./models:/opt/comfy_ui/models
+    - models:/opt/comfy_ui/models
-    - ./output:/opt/comfy_ui/output
+    - output:/opt/comfy_ui/output
-    - ./input:/opt/comfy_ui/input
+    - input:/opt/comfy_ui/input
-    - ./custom_nodes:/opt/comfy_ui/custom_nodes
+    - custom_nodes:/opt/comfy_ui/custom_nodes
+  restart: unless-stopped
 
 x-nvidia_docker: &nvidia_docker
   deploy:
@@ -26,6 +29,10 @@ x-nvidia_docker: &nvidia_docker
             capabilities: [gpu]
 
 x-base_service_amd: &amd_docker
+  build:
+    context: docker_files
+    dockerfile: amd.Dockerfile
+
   group_add:
     - video
   devices:
@@ -58,3 +65,10 @@ services:
     profiles: ["cpu"]
     environment:
       - CLI_ARGS=--listen --port 8188 --cpu
+
+
+volumes:
+  models:
+  output:
+  input:
+  custom_nodes:

docker/docker_files/amd.Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.10-slim
+
+
+WORKDIR /opt/comfy_ui
+
+
+RUN PIP_NO_CACHE_DIR=1 pip install --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 \
+torch \
+torchvision \
+torchaudio
+
+COPY ../.. /opt/comfy_ui
+RUN pip install -r requirements.txt
+
+
+ENV CLI_ARGS=""
+CMD python main.py ${CLI_ARGS}

docker/docker_files/default.Dockerfile

@@ -11,7 +11,7 @@ torchaudio \
 xformers \
 triton
 
-COPY . /opt/comfy_ui
+COPY ../.. /opt/comfy_ui
 RUN pip install -r requirements.txt