Upload Dockerfile
Browse files- 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 |
|