diff --git a/StableDiffusion.ipynb b/StableDiffusion.ipynb
index 8ae2dcb..20d6b59 100644
--- a/StableDiffusion.ipynb
+++ b/StableDiffusion.ipynb
@@ -42,6 +42,7 @@ async def main():
   !dpkg -i *.deb
   %env LD_PRELOAD=libtcmalloc.so
   !rm *.deb
+  !pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116 -U
   await async_command_group([
     "apt -y install -qq aria2",
     "pip install -q --pre xformers==0.0.17.dev476 -U",