diff --git a/docker-compose.yml b/docker-compose.yml
index 6692a99..71df7cc 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -51,7 +51,7 @@ services:
     <<: *base_service
     profiles: ["lstein"]
     build: ./services/lstein/
-    image: sd-lstein:1
+    image: sd-lstein:2
     environment:
       - PRELOAD=true
       - CLI_ARGS=
diff --git a/services/lstein/Dockerfile b/services/lstein/Dockerfile
index 1c20222..163c32c 100644
--- a/services/lstein/Dockerfile
+++ b/services/lstein/Dockerfile
@@ -24,7 +24,7 @@ conda clean -a -y
 EOF
 
 
-ARG BRANCH=development SHA=554445a985d970200095bbcb109273a49c462682
+ARG BRANCH=development SHA=3081b6b7dd4c2fb1156e7a99dc461012c4ecda35
 RUN <<EOF
 cd stable-diffusion
 git fetch