[CONF] changed dockerfile to enable nvidia gpu usage

Dockerfile

@@ -1,4 +1,4 @@
-FROM ubuntu:22.04
+FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive \
     TZ=Europe/Paris
@@ -48,28 +48,29 @@ ADD ./workflows ./user/default/workflows
 # Checkpoints
 RUN echo "Downloading checkpoints..."
 # Quiet because it's a big file (2.13 GB)
-# RUN wget -c https://huggingface.co/genai-archive/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.fp16.safetensors -P ./models/checkpoints/ --quiet
+RUN wget -c https://huggingface.co/genai-archive/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.fp16.safetensors -P ./models/checkpoints/ --quiet
 
 # VAE
 RUN echo "Downloading VAE..."
-# RUN wget -c https://huggingface.co/black-forest-labs/FLUX.1-schnell/resolve/main/ae.safetensors -P ./models/vae/
+RUN wget -c https://huggingface.co/black-forest-labs/FLUX.1-schnell/resolve/main/ae.safetensors -P ./models/vae/
 
 # Loras
 RUN echo "Downloading Loras..."
-# RUN wget -c https://huggingface.co/2ndChanceParis/industrial-lora/resolve/main/industrial_lora-07.safetensors -P ./models/loras
+RUN wget -c https://huggingface.co/2ndChanceParis/industrial-lora/resolve/main/industrial_lora-07.safetensors -P ./models/loras
 
 # CLIP model
 RUN echo "Downloading CLIP models..."
 # Quiet because it's a big file (9.79 GB)
-# RUN wget -c https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/t5xxl_fp16.safetensors -P ./models/clip/ --quiet
-# RUN wget -c https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/clip_l.safetensors -P ./models/clip/
+RUN wget -c https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/t5xxl_fp16.safetensors -P ./models/clip/ --quiet
+RUN wget -c https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/clip_l.safetensors -P ./models/clip/
 
 # Unet model
-# RUN echo "Downloading Unet models..."
+RUN echo "Downloading Unet models..."
 # Quiet because it's a big file (23.8 GB)
-# RUN wget -c https://huggingface.co/camenduru/FLUX.1-dev/resolve/fc63f3204a12362f98c04bc4c981a06eb9123eee/flux1-canny-dev.safetensors -P ./models/unet/ --quiet
+RUN wget -c https://huggingface.co/camenduru/FLUX.1-dev/resolve/fc63f3204a12362f98c04bc4c981a06eb9123eee/flux1-canny-dev.safetensors -P ./models/unet/ --quiet
 
 # --- Optional models ---
+# Not used at the moment
 
 # T2I-Adapter
 # RUN echo "Downloading T2I-Adapters..."
@@ -120,4 +121,4 @@ RUN cd custom_nodes && git clone https://github.com/ltdrdata/ComfyUI-Manager.git
 
 RUN echo "Done"
 
-CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--port", "7860", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]
+CMD ["python", "main.py", "--listen", "0.0.0.0", "--port", "7860", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]

README.md

@@ -8,4 +8,32 @@ pinned: false
 short_description: ComfyUI API - Docker
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Comfy UI Api
+
+This is a simple example of how to run ComfyUI on a nvidia cuda machine using Docker.
+
+## Requirements
+
+- docker & docker-compose
+- nvidia gpu
+
+## Installation
+
+### Local
+
+1. Clone the repository
+2. Run `docker compose up` in the root directory of the repository
+3. Open a browser and navigate to :
+    - ComfyUI: `http://localhost:7860`
+
+### Hugging Face Spaces
+
+When pushing the repository to Hugging Face Spaces, the container is automatically built and started from the
+root [Dockerfile](Dockerfile).
+Hugging Face Spaces exposes port 7860 on which ComfyUI is running after build and startup.
+
+## Usage
+
+### Workflows
+
+Workflows stored in [workflows](workflows) are copied to the container.

docker-compose.yml

@@ -3,14 +3,14 @@ version: "3"
 services:
   comfyui:
     build: .
-#    environment:
-#          - NVIDIA_VISIBLE_DEVICES=all
+    environment:
+      - NVIDIA_VISIBLE_DEVICES=all
     ports:
       - "7860:7860"
-#    deploy:
-#      resources:
-#        reservations:
-#          devices:
-#            - driver: nvidia
-#              count: 1
-#              capabilities: [gpu]
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [ gpu ]