wrdias commited on
Commit
1e82fe1
·
verified ·
1 Parent(s): a95e569

Upload Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
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
@@ -236,6 +237,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
 
 
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
 
237
  RUN echo "Done"
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