Upload 2 files

easydiffusion/_main.ipynb

@@ -54,22 +54,18 @@
     "\n",
     "fix_script()\n",
     "os.chdir(\"sd-ui-files\")\n",
-    "!python ./scripts/check_modules.py  #安装依赖&环境\n",
+    "!python current_dir/scripts/check_modules.py  #安装依赖&环境\n",
     "!pip install --upgrade pytorch-lightning\n",
     "\n",
-    "!curl -Lo /content/models/stable-diffusion/chilloutmix_NiPrunedFp32Fix-chonghui.safetensors https://huggingface.co/spaces/weo1101/111/resolve/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting.safetensors\n",
-    "!curl -Lo /content/models/stable-diffusion/chilloutmix_NiPrunedFp32Fix.safetensors https://huggingface.co/spaces/weo1101/111/resolve/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting.safetensors\n",
-    "\n",
-    "# !curl -Lo content/apply_func.py  https://github.com/Van-wise/sd-colab/raw/main/easydiffusion/apply_func.py\n",
-    "# !rm -rf /usr/local/lib/python3.10/dist-packages/pytorch_lightning/utilities/apply_func.py\n",
-    "# !cp /content/apply_func.py /usr/local/lib/python3.10/dist-packages/pytorch_lightning/utilities/apply_func.py\n",
+    "!curl -Lo {current_dir}/models/stable-diffusion/chilloutmix_NiPrunedFp32Fix-chonghui.safetensors https://huggingface.co/spaces/weo1101/111/resolve/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting.safetensors\n",
+    "!curl -Lo {current_dir}/models/stable-diffusion/chilloutmix_NiPrunedFp32Fix.safetensors https://huggingface.co/spaces/weo1101/111/resolve/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting.safetensors\n",
     "\n",
     "from pathlib import Path\n",
     "os.environ[\"SD_UI_PATH\"] = str(Path(Path.cwd(), \"ui\"))\n",
     "os.environ[\"INSTALL_ENV_DIR\"] = \"/usr/local\"\n",
     "print(str(Path(Path.cwd(), \"ui\")))\n",
     "\n",
-    "!python ./scripts/check_modules.py --launch-uvicorn"
+    "print(\"Installing Easy Diffusion's UI..\\n\")"
    ]
   }
  ],

easydiffusion/_start.ipynb

@@ -9,11 +9,6 @@
     "import threading\n",
     "import time\n",
     "import os\n",
-    "if not os.path.exists('/content/key/id_rsa'):\n",
-    "  !mkdir -p /content/key/id_rsa\n",
-    "  !ssh-keygen -t rsa -b 4096 -N \"\" -f /content/key/id_rsa -y\n",
-    "  !chmod 644 /content/key/id_rsa\n",
-    "  !chmod 644 /content/key/id_rsa.pub\n",
     "\n",
     "def tunnel():\n",
     "  time.sleep(1)\n",
@@ -24,12 +19,11 @@
     "  print(ngrok_tunnel)\n",
     "  print(cloudflare_url)\n",
     "  time.sleep(1)\n",
-    "  !ssh -R 80:127.0.0.1:9000 -o StrictHostKeyChecking=no -i /content/key/id_rsa remote.moe\n",
     "\n",
     "threading.Thread(target=tunnel, daemon=True).start()\n",
     "\n",
-    "!python ./scripts/check_modules.py --launch-uvicorn\n",
-    "#!python ./scripts//ui/main.py"
+    "!python {current_dir}/scripts/check_modules.py --launch-uvicorn\n",
+    "#!python ./ui/main.py"
    ]
   }
  ],