Merge pull request #10 from HenriZuber/main

fix: torchmetrics import issue
camenduru · Aug 26, 2023 · 041daad · 041daad
2 parents 44fb2eb + fc117fe
commit 041daad
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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