Spaces:
Running
Running
Upload Dockerfile
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
@@ -58,6 +58,7 @@ RUN pyenv install $PYTHON_VERSION && \
|
|
58 |
pip install addict && \
|
59 |
pip install pynvml && \
|
60 |
pip install yapf && \
|
|
|
61 |
pip install timm
|
62 |
#pip install huggingface_hub["cli"] && \
|
63 |
#huggingface-cli delete-cache
|
@@ -237,5 +238,5 @@ RUN echo "Downloading checkpoints..." && \
|
|
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 |
|
|
|
58 |
pip install addict && \
|
59 |
pip install pynvml && \
|
60 |
pip install yapf && \
|
61 |
+
pip install numexpr && \
|
62 |
pip install timm
|
63 |
#pip install huggingface_hub["cli"] && \
|
64 |
#huggingface-cli delete-cache
|
|
|
238 |
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
|
239 |
|
240 |
#CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]
|
241 |
+
CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--output-directory", "--multi-user CLI"]
|
242 |
|