wrdias commited on
Commit
451ca25
·
verified ·
1 Parent(s): de0c2f2

Upload Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -224,6 +224,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
 
 
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
+ RUN mkdir /home/user/app/models/insightface
228
+
229
  #CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]
230
  CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--cpu", "--port", "7860", "--use-split-cross-attention", "--multi-user"]
231