diff --git a/stable-diffusion/setup-lamacleaner-paperspace.sh b/stable-diffusion/setup-lamacleaner-paperspace.sh
index 9beab6a..288c1c4 100644
--- a/stable-diffusion/setup-lamacleaner-paperspace.sh
+++ b/stable-diffusion/setup-lamacleaner-paperspace.sh
@@ -8,7 +8,8 @@ cd /tmp
 python -m venv --without-pip /tmp/lama-venv
 wget https://bootstrap.pypa.io/get-pip.py
 /tmp/lama-venv/bin/python get-pip.py
-/tmp/lama-venv/bin/pip install torch==1.13.1+cu117 \
-    torchvision==0.14.1 triton xformers==0.0.16rc425 \
-        --extra-index-url https://download.pytorch.org/whl/cu117
+/tmp/lama-venv/bin/pip install torch==2.0.1+cu118 torchvision triton --extra-index-url https://download.pytorch.org/whl/cu118
+/tmp/lama-venv/bin/pip install xformers==0.0.20 --no-deps
+/tmp/lama-venv/bin/pip install pyre-extensions==0.0.29
 /tmp/lama-venv/bin/pip install lama-cleaner
+/tmp/lama-venv/bin/pip install rembg