Skip to content

Commit b4c9713

Browse files
committed
add more models
1 parent 617465d commit b4c9713

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

Dockerfile

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ ARG MODEL_TYPE
6464
WORKDIR /comfyui
6565

6666
# Create necessary directories
67-
RUN mkdir -p models/checkpoints models/vae models/facedetection models/facerestore_models models/facexlib models/insightface models/onnx
67+
RUN mkdir -p models/checkpoints models/vae models/facedetection models/facerestore_models models/facexlib models/insightface models/onnx models/reswapper
6868

6969
# Download checkpoints/vae/LoRA to include in image based on model type
7070
RUN if [ "$MODEL_TYPE" = "sdxl" ]; then \
@@ -95,6 +95,8 @@ RUN wget -O models/facerestore_models/GFPGANv1.3.pth https://huggingface.co/data
9595
wget -O models/insightface/inswapper_128.onnx https://huggingface.co/datasets/Gourieff/ReActor/resolve/main/models/inswapper_128.onnx && \
9696
wget -O models/reswapper/reswapper_256.onnx https://huggingface.co/datasets/Gourieff/ReActor/resolve/main/models/reswapper_256.onnx && \
9797
wget -O models/facedetection/detection_Resnet50_Final.pth https://huggingface.co/darkeril/collection/resolve/main/detection_Resnet50_Final.pth && \
98+
wget -O models/facexlib/detection_Resnet50_Final.pth https://huggingface.co/darkeril/collection/resolve/main/detection_Resnet50_Final.pth && \
99+
wget -O models/facexlib/parsing_bisenet.pth https://huggingface.co/caocaocoa/1111/resolve/4f49a96a8919398af6e6373ed7dd6e323fefcdb8/parsing_bisenet.pth && \
98100
wget -O models/facedetection/parsing_parsenet.pth https://huggingface.co/gmk123/GFPGAN/resolve/main/parsing_parsenet.pth && \
99101
wget -O models/facedetection/yolov5l-face.pth https://huggingface.co/martintomov/comfy/resolve/main/facedetection/yolov5l-face.pth
100102

0 commit comments

Comments
 (0)