更新 'StableDiffusion.ipynb'

2023-03-22 01:07:11 +00:00 · 2023-03-22 01:07:11 +00:00 · 9366c52229
commit 9366c52229
parent e0ff9b2f70
1 changed files with 1 additions and 0 deletions
--- 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",