lemonteaa commited on
Commit
f744ccf
·
1 Parent(s): 5bee4a5

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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/"]