From 9aea8abe1375f3d48aa9742c444b5242111e3121 Mon Sep 17 00:00:00 2001 From: TimPietrusky Date: Fri, 17 Nov 2023 17:11:36 +0100 Subject: [PATCH] feat: added xl_more_art-full_v1, improved comments --- Dockerfile | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/Dockerfile b/Dockerfile index 1f85df1..b86c3d7 100644 --- a/Dockerfile +++ b/Dockerfile @@ -32,13 +32,15 @@ RUN pip3 install --no-cache-dir torch torchvision torchaudio --index-url https:/ # Install runpod RUN pip3 install runpod requests -# Download models to include in image. (not required if including other models below) +# 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/checkpoints/sdxl_vae.safetensors https://huggingface.co/stabilityai/sdxl-vae/resolve/main/sdxl_vae.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/lora/xl_more_art-full_v1.safetensors https://civitai.com/api/download/models/152309 -# # Example for adding specific models into image +# Example for adding specific models into image # ADD models/checkpoints/sd_xl_base_1.0.safetensors models/checkpoints/ -# ADD models/checkpoints/sdxl_vae.safetensors models/checkpoints/ +# ADD models/vae/sdxl_vae.safetensors models/vae/ # Go back to the root WORKDIR /