wrdias commited on
Commit
6827ff6
·
verified ·
1 Parent(s): 5625aef

Upload Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -0
Dockerfile CHANGED
@@ -53,6 +53,7 @@ RUN pyenv install $PYTHON_VERSION && \
53
  pip install gray2color && \
54
  pip install segment-anything && \
55
  pip install typing-aliases && \
 
56
  pip install insightface && \
57
  pip install deepdiff && \
58
  pip install addict && \
@@ -224,6 +225,8 @@ RUN echo "Downloading checkpoints..." && \
224
  RUN echo "Done"
225
  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
226
 
 
 
227
  #CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]
228
  CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--multi-user"]
229
 
 
53
  pip install gray2color && \
54
  pip install segment-anything && \
55
  pip install typing-aliases && \
56
+ pip install typing-extensions && \
57
  pip install insightface && \
58
  pip install deepdiff && \
59
  pip install addict && \
 
225
  RUN echo "Done"
226
  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
227
 
228
+ RUN mkdir /home/user/app/models/insightface
229
+
230
  #CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]
231
  CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--multi-user"]
232