Spaces:
Running
Running
Upload Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
@@ -236,6 +236,6 @@ RUN echo "Downloading checkpoints..." && \
|
|
236 |
RUN echo "Done"
|
237 |
RUN pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
|
238 |
|
239 |
-
CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]
|
240 |
-
|
241 |
|
|
|
236 |
RUN echo "Done"
|
237 |
RUN pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
|
238 |
|
239 |
+
#CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]
|
240 |
+
CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--output-directory"]
|
241 |
|