From de0b4e7eb083e0df609cf4e58515bbb595e65595 Mon Sep 17 00:00:00 2001 From: Xiujuan Li Date: Wed, 13 Dec 2023 16:04:55 +0800 Subject: [PATCH] feat: test cn docker image --- build_scripts/Dockerfile.inference.cn.from_scratch | 3 +-- build_scripts/Dockerfile.utils.cn.from_scratch | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/build_scripts/Dockerfile.inference.cn.from_scratch b/build_scripts/Dockerfile.inference.cn.from_scratch index 766b2daa..5de65f28 100644 --- a/build_scripts/Dockerfile.inference.cn.from_scratch +++ b/build_scripts/Dockerfile.inference.cn.from_scratch @@ -33,8 +33,7 @@ RUN apt-get install -y file && \ trash /opt/conda/lib/python3.10/site-packages/opencv_python* # prepare environment fow webui -RUN mkdir /opt/ml/code/models/Stable-diffusion && \ - wget -O https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.safetensors /opt/ml/code/models/Stable-diffusion +RUN wget -O https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.safetensors /opt/ml/code/models/Stable-diffusion RUN pip install -r /opt/ml/code/requirements_versions.txt && \ pip install https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip && \ pip install https://github.com/mlfoundations/open_clip/archive/bb6e834e9c70d9c27d0dc3ecedeebeaeb1ffad6b.zip && \ diff --git a/build_scripts/Dockerfile.utils.cn.from_scratch b/build_scripts/Dockerfile.utils.cn.from_scratch index 8f7bc847..688607a0 100644 --- a/build_scripts/Dockerfile.utils.cn.from_scratch +++ b/build_scripts/Dockerfile.utils.cn.from_scratch @@ -33,8 +33,7 @@ RUN pip install -r /opt/ml/code/extensions/sd_dreambooth_extension/requirements. apt-get install -y google-perftools # prepare environment fow webui -RUN mkdir /opt/ml/code/models/Stable-diffusion && \ - wget -O https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.safetensors /opt/ml/code/models/Stable-diffusion +RUN wget -O https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.safetensors /opt/ml/code/models/Stable-diffusion RUN pip install -r /opt/ml/code/requirements_versions.txt && \ pip install https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip && \ pip install https://github.com/mlfoundations/open_clip/archive/bb6e834e9c70d9c27d0dc3ecedeebeaeb1ffad6b.zip && \