fix

Dockerfile

@@ -3,8 +3,6 @@ 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 \
@@ -142,7 +140,7 @@ RUN cd custom_nodes && git clone https://github.com/EllangoK/ComfyUI-post-proces
 
 RUN echo "Done"
 
-CMD ["python", "main.py", "--listen", "0.0.0.0", "--port", "7860", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]
+CMD ["python", "main.py", "--listen", "0.0.0.0", "--port", "7860"]