radames commited on
Commit
50b090f
·
1 Parent(s): 103399c

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +20 -18
Dockerfile CHANGED
@@ -12,7 +12,7 @@ RUN apt-get update && apt-get install -y \
12
  libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
13
  libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev git-lfs \
14
  ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
15
- nginx netcat-traditional \
16
  && rm -rf /var/lib/apt/lists/* \
17
  && git lfs install
18
 
@@ -57,10 +57,13 @@ RUN pyenv install $PYTHON_VERSION && \
57
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
58
 
59
  # Set the working directory to /data if USE_PERSISTENT_DATA is set, otherwise set to $HOME/app
60
- WORKDIR $HOME/app
61
 
62
- RUN chown -R user:user $HOME/app
 
 
63
 
 
64
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
65
 
66
  RUN git clone https://github.com/comfyanonymous/ComfyUI . && \
@@ -70,13 +73,13 @@ RUN git clone https://github.com/comfyanonymous/ComfyUI . && \
70
 
71
  RUN echo "Downloading checkpoints..."
72
  # SDXL
73
- ADD https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors ./models/checkpoints/
74
- ADD https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0/resolve/main/sd_xl_refiner_1.0.safetensors ./models/checkpoints/
75
  # ADD https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0_0.9vae.safetensors ./models/checkpoints/
76
 
77
  # SD1.5
78
- # ADD https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt ./models/checkpoints/
79
- ADD https://huggingface.co/stabilityai/stable-diffusion-2-1-base/resolve/main/v2-1_512-ema-pruned.safetensors ./models/checkpoints/
80
  # ADD https://huggingface.co/stabilityai/stable-diffusion-2-1/resolve/main/v2-1_768-ema-pruned.safetensors ./models/checkpoints/
81
  # Some SD1.5 anime style
82
  # ADD https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/Models/AbyssOrangeMix2/AbyssOrangeMix2_hard.safetensors ./models/checkpoints/
@@ -114,7 +117,7 @@ ADD https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-f
114
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_canny_fp16.safetensors ./models/controlnet/
115
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11f1p_sd15_depth_fp16.safetensors ./models/controlnet/
116
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_inpaint_fp16.safetensors ./models/controlnet/
117
- ADD https://huggingface.co/thibaud/controlnet-sd21/resolve/main/control_v11p_sd21_lineart.safetensors ./models/controlnet/
118
  #ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_lineart_fp16.safetensors ./models/controlnet/
119
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_mlsd_fp16.safetensors ./models/controlnet/
120
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_normalbae_fp16.safetensors ./models/controlnet/
@@ -124,18 +127,18 @@ ADD https://huggingface.co/thibaud/controlnet-sd21/resolve/main/control_v11p_sd2
124
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_softedge_fp16.safetensors ./models/controlnet/
125
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15s2_lineart_anime_fp16.safetensors ./models/controlnet/
126
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11u_sd15_tile_fp16.safetensors ./models/controlnet/
127
- ADD https://huggingface.co/thibaud/controlnet-openpose-sdxl-1.0/resolve/main/OpenPoseXL2.safetensors ./models/controlnet/
128
 
129
  # https://huggingface.co/stabilityai/control-lora
130
- ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank256/control-lora-canny-rank256.safetensors ./models/controlnet/
131
- ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank256/control-lora-depth-rank256.safetensors ./models/controlnet/
132
- ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank256/control-lora-recolor-rank256.safetensors ./models/controlnet/
133
- ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank256/control-lora-sketch-rank256.safetensors ./models/controlnet/
134
 
135
- ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank128/control-lora-canny-rank128.safetensors ./models/controlnet/
136
- ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank128/control-lora-depth-rank128.safetensors ./models/controlnet/
137
- ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank128/control-lora-recolor-rank128.safetensors ./models/controlnet/
138
- ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank128/control-lora-sketch-rank128-metadata.safetensors ./models/controlnet/
139
 
140
 
141
  # ADD https://huggingface.co/thibaud/controlnet-openpose-sdxl-1.0/resolve/main/diffusion_pytorch_model.bin -O ./models/controlnet/OpenPoseXL2.bin
@@ -159,7 +162,6 @@ RUN cd custom_nodes && git clone https://github.com/EllangoK/ComfyUI-post-proces
159
  RUN cd custom_nodes && git clone https://github.com/ltdrdata/ComfyUI-Manager.git
160
 
161
  RUN echo "Done"
162
- RUN chown -R user:user $HOME/app
163
 
164
  COPY --chown=user run.sh .
165
 
 
12
  libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
13
  libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev git-lfs \
14
  ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
15
+ nginx netcat \
16
  && rm -rf /var/lib/apt/lists/* \
17
  && git lfs install
18
 
 
57
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
58
 
59
  # Set the working directory to /data if USE_PERSISTENT_DATA is set, otherwise set to $HOME/app
60
+ WORKDIR $HOME/app
61
 
62
+ USER root
63
+ RUN chown -R user:user $HOME/app \
64
+ && chmod -R 755 $HOME/app
65
 
66
+ USER user
67
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
68
 
69
  RUN git clone https://github.com/comfyanonymous/ComfyUI . && \
 
73
 
74
  RUN echo "Downloading checkpoints..."
75
  # SDXL
76
+ # ADD https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors ./models/checkpoints/
77
+ # ADD https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0/resolve/main/sd_xl_refiner_1.0.safetensors ./models/checkpoints/
78
  # ADD https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0_0.9vae.safetensors ./models/checkpoints/
79
 
80
  # SD1.5
81
+ ADD https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt ./models/checkpoints/
82
+ # ADD https://huggingface.co/stabilityai/stable-diffusion-2-1-base/resolve/main/v2-1_512-ema-pruned.safetensors ./models/checkpoints/
83
  # ADD https://huggingface.co/stabilityai/stable-diffusion-2-1/resolve/main/v2-1_768-ema-pruned.safetensors ./models/checkpoints/
84
  # Some SD1.5 anime style
85
  # ADD https://huggingface.co/WarriorMama777/OrangeMixs/resolve/main/Models/AbyssOrangeMix2/AbyssOrangeMix2_hard.safetensors ./models/checkpoints/
 
117
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_canny_fp16.safetensors ./models/controlnet/
118
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11f1p_sd15_depth_fp16.safetensors ./models/controlnet/
119
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_inpaint_fp16.safetensors ./models/controlnet/
120
+ # ADD https://huggingface.co/thibaud/controlnet-sd21/resolve/main/control_v11p_sd21_lineart.safetensors ./models/controlnet/
121
  #ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_lineart_fp16.safetensors ./models/controlnet/
122
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_mlsd_fp16.safetensors ./models/controlnet/
123
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_normalbae_fp16.safetensors ./models/controlnet/
 
127
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15_softedge_fp16.safetensors ./models/controlnet/
128
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11p_sd15s2_lineart_anime_fp16.safetensors ./models/controlnet/
129
  # ADD https://huggingface.co/comfyanonymous/ControlNet-v1-1_fp16_safetensors/resolve/main/control_v11u_sd15_tile_fp16.safetensors ./models/controlnet/
130
+ # ADD https://huggingface.co/thibaud/controlnet-openpose-sdxl-1.0/resolve/main/OpenPoseXL2.safetensors ./models/controlnet/
131
 
132
  # https://huggingface.co/stabilityai/control-lora
133
+ # ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank256/control-lora-canny-rank256.safetensors ./models/controlnet/
134
+ # ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank256/control-lora-depth-rank256.safetensors ./models/controlnet/
135
+ # ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank256/control-lora-recolor-rank256.safetensors ./models/controlnet/
136
+ # ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank256/control-lora-sketch-rank256.safetensors ./models/controlnet/
137
 
138
+ # ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank128/control-lora-canny-rank128.safetensors ./models/controlnet/
139
+ # ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank128/control-lora-depth-rank128.safetensors ./models/controlnet/
140
+ # ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank128/control-lora-recolor-rank128.safetensors ./models/controlnet/
141
+ # ADD https://huggingface.co/stabilityai/control-lora/resolve/main/control-LoRAs-rank128/control-lora-sketch-rank128-metadata.safetensors ./models/controlnet/
142
 
143
 
144
  # ADD https://huggingface.co/thibaud/controlnet-openpose-sdxl-1.0/resolve/main/diffusion_pytorch_model.bin -O ./models/controlnet/OpenPoseXL2.bin
 
162
  RUN cd custom_nodes && git clone https://github.com/ltdrdata/ComfyUI-Manager.git
163
 
164
  RUN echo "Done"
 
165
 
166
  COPY --chown=user run.sh .
167