diff --git a/runpod/serverless-automatic/Dockerfile b/runpod/serverless-automatic/Dockerfile index 230dde6..e19275f 100644 --- a/runpod/serverless-automatic/Dockerfile +++ b/runpod/serverless-automatic/Dockerfile @@ -30,6 +30,11 @@ RUN mkdir /workspace/stable-diffusion-webui/models/Codeformer/ && \ wget -O /workspace/stable-diffusion-webui/models/ESRGAN/ESRGAN.pth https://github.com/cszn/KAIR/releases/download/v1.0/ESRGAN.pth && \ # mkdir /workspace/stable-diffusion-webui/models/Stable-diffusion/tokenizer/ && \ # wget -O /workspace/stable-diffusion-webui/models/Stable-diffusion/tokenizer/vocab.json https://huggingface.co/runwayml/stable-diffusion-v1-5/raw/main/tokenizer/vocab.json && \ + mkdir /workspace/stable-diffusion-webui/models/Real-ESRGAN/ && \ + wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth -P /workspace/stable-diffusion-webui/models/Real-ESRGAN/ && \ + wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.1/RealESRNet_x4plus.pth -P /workspace/stable-diffusion-webui/models/Real-ESRGAN/ && \ + wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth -P /workspace/stable-diffusion-webui/models/Real-ESRGAN/ && \ + wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth -P /workspace/stable-diffusion-webui/models/Real-ESRGAN/ && \ git clone --depth 1 https://github.com/Coyote-A/ultimate-upscale-for-automatic1111 /workspace/stable-diffusion-webui/extensions/ultimate-upscale-for-automatic1111 RUN pip install runpod