wrdias commited on
Commit
da88923
·
verified ·
1 Parent(s): 5787ba8

Upload Dockerfile

Browse files
Files changed (1) hide show
  1. 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