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 |
|