new

Dockerfile

@@ -3,6 +3,8 @@ FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
 ENV DEBIAN_FRONTEND=noninteractive \
     TZ=America/Los_Angeles
 
+ARG USE_PERSISTENT_DATA
+
 RUN apt-get update && apt-get install -y \
     git \
     make build-essential libssl-dev zlib1g-dev \
@@ -155,7 +157,7 @@ RUN cd custom_nodes && git clone https://github.com/ltdrdata/ComfyUI-Manager.git
 
 RUN echo "Done"
 
-CMD ["python", "main.py", "--listen", "0.0.0.0", "--port", "7860"]
+CMD ["python", "main.py", "--listen", "0.0.0.0", "--port", "7860", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]