Update Dockerfile
Browse files- Dockerfile +11 -6
Dockerfile
CHANGED
@@ -15,14 +15,19 @@ RUN mkdir /tmp/comfy_out
|
|
15 |
|
16 |
RUN git clone https://github.com/comfyanonymous/ComfyUI
|
17 |
WORKDIR /home/hf/ComfyUI
|
|
|
|
|
|
|
|
|
|
|
18 |
RUN pip install -r requirements.txt
|
19 |
|
20 |
-
RUN wget -c https://huggingface.co/stabilityai/sdxl-turbo/resolve/main/sd_xl_turbo_1.0_fp16.safetensors -P ./models/checkpoints/
|
21 |
-
RUN wget -c https://huggingface.co/openskyml/lcm-lora-sdxl-turbo/resolve/main/lcm-lora-sdxl-turbo.safetensors -P ./models/loras/
|
22 |
-
RUN wget -c https://huggingface.co/kataragi/SDXL-turbo-LoRA/resolve/main/sd_xl_turbo_lora_rank64.safetensors -P ./models/loras/
|
23 |
-
RUN wget -c https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors -P ./models/vae/
|
24 |
-
RUN wget -c https://huggingface.co/sberbank-ai/Real-ESRGAN/resolve/main/RealESRGAN_x2.pth -P ./models/upscale_models/
|
25 |
|
26 |
-
RUN pip install safetensors diffusers
|
27 |
|
28 |
CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--port", "7860", "--output-directory", "/tmp/comfy_out/"]
|
|
|
15 |
|
16 |
RUN git clone https://github.com/comfyanonymous/ComfyUI
|
17 |
WORKDIR /home/hf/ComfyUI
|
18 |
+
|
19 |
+
ENV VIRTUAL_ENV=/home/hf/ComfyUI/venv
|
20 |
+
RUN python3 -m venv $VIRTUAL_ENV
|
21 |
+
ENV PATH="$VIRTUAL_ENV/bin:$PATH"
|
22 |
+
|
23 |
RUN pip install -r requirements.txt
|
24 |
|
25 |
+
RUN wget -nv -c https://huggingface.co/stabilityai/sdxl-turbo/resolve/main/sd_xl_turbo_1.0_fp16.safetensors -P ./models/checkpoints/
|
26 |
+
RUN wget -nv -c https://huggingface.co/openskyml/lcm-lora-sdxl-turbo/resolve/main/lcm-lora-sdxl-turbo.safetensors -P ./models/loras/
|
27 |
+
RUN wget -nv -c https://huggingface.co/kataragi/SDXL-turbo-LoRA/resolve/main/sd_xl_turbo_lora_rank64.safetensors -P ./models/loras/
|
28 |
+
RUN wget -nv -c https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors -P ./models/vae/
|
29 |
+
RUN wget -nv -c https://huggingface.co/sberbank-ai/Real-ESRGAN/resolve/main/RealESRGAN_x2.pth -P ./models/upscale_models/
|
30 |
|
31 |
+
#RUN pip install safetensors diffusers
|
32 |
|
33 |
CMD ["python", "main.py", "--cpu", "--listen", "0.0.0.0", "--port", "7860", "--output-directory", "/tmp/comfy_out/"]
|