From fc117fe403c84bdac209f6af958ca4d8a83441d5 Mon Sep 17 00:00:00 2001 From: HenriZuber Date: Tue, 18 Jul 2023 11:35:42 +0200 Subject: [PATCH] fix: torchmetrics import issue --- Dockerfile.Lite | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile.Lite b/Dockerfile.Lite index 0147e18..051ab71 100644 --- a/Dockerfile.Lite +++ b/Dockerfile.Lite @@ -33,6 +33,7 @@ RUN sed -i -e '''/ prepare_environment()/a\ os.system\(f\"""sed -i -e ''\" RUN sed -i -e 's/fastapi==0.90.1/fastapi==0.89.1/g' /content/stable-diffusion-webui/requirements_versions.txt RUN sed -i -e 's/ start()/ #start()/g' /content/stable-diffusion-webui/launch.py RUN cd stable-diffusion-webui && python launch.py --skip-torch-cuda-test +RUN pip install torchmetrics==0.11.4 ADD --chown=user https://huggingface.co/ckpt/sd15/resolve/main/v1-5-pruned-emaonly.ckpt /content/stable-diffusion-webui/models/Stable-diffusion/v1-5-pruned-emaonly.ckpt