diff --git a/Dockerfile b/Dockerfile index 699a340d..01409b4c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -34,7 +34,9 @@ RUN pip3 install runpod requests # Download checkpoints/vae/LoRA to include in image RUN wget -O models/checkpoints/sd_xl_base_1.0.safetensors https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors +RUN wget -O models/vae/sdxl_vae.safetensors https://huggingface.co/stabilityai/sdxl-vae/resolve/main/sdxl_vae.safetensors RUN wget -O models/vae/sdxl-vae-fp16-fix.safetensors https://huggingface.co/madebyollin/sdxl-vae-fp16-fix/resolve/main/sdxl_vae.safetensors +RUN wget -O models/loras/xl_more_art-full_v1.safetensors https://civitai.com/api/download/models/152309 # Example for adding specific models into image # ADD models/checkpoints/sd_xl_base_1.0.safetensors models/checkpoints/