diff --git a/services/lstein/Dockerfile b/services/lstein/Dockerfile index 5063747..8464c6c 100644 --- a/services/lstein/Dockerfile +++ b/services/lstein/Dockerfile @@ -23,7 +23,7 @@ conda clean -a -y EOF -ARG BRANCH=development SHA=50d607ffea3734072a80e38b09ba0c3758af5d40 +ARG BRANCH=development SHA=b40bfb5116b7fc618f78a0d152005ceb46153443 # this breaks on generation: # there is a new UI anyway, but it is not by any means ready. # ARG BRANCH=development SHA=bdbc76fcd4bd3362312dc91b087d9af66de423b1 diff --git a/services/lstein/mount.sh b/services/lstein/mount.sh index bae7a0b..0b31bdf 100755 --- a/services/lstein/mount.sh +++ b/services/lstein/mount.sh @@ -7,12 +7,11 @@ ROOT=/stable-diffusion mkdir -p "${ROOT}/models/ldm/stable-diffusion-v1/" ln -sf /cache/models/model.ckpt "${ROOT}/models/ldm/stable-diffusion-v1/model.ckpt" -if test -f /cache/models/GFPGANv1.3.pth; then - base="${ROOT}/src/gfpgan/experiments/pretrained_models/" - mkdir -p "${base}" - ln -sf /cache/models/GFPGANv1.3.pth "${base}/GFPGANv1.3.pth" - echo "Mounted GFPGANv1.3.pth" -fi +base="${ROOT}/src/gfpgan/experiments/pretrained_models/" +mkdir -p "${base}" +# TODO: "real" GFPGANv1.4.pth +ln -sf /cache/models/GFPGANv1.3.pth "${base}/GFPGANv1.4.pth" +echo "Mounted GFPGANv1.3.pth" # facexlib FACEX_WEIGHTS=/opt/conda/lib/python3.9/site-packages/facexlib/weights