Upload 3 files

easydiffusion/_main.ipynb

@@ -31,8 +31,6 @@
     "    else:\n",
     "        print(\"git clone failed\")\n",
     "\n",
-    "os.chdir(\"sd-ui-files\")\n",
-    "\n",
     "def fix_script():\n",
     "    file_path = f\"{current_dir}/sd-ui-files/scripts/check_modules.py\"\n",
     "\n",
@@ -55,23 +53,22 @@
     "    print(\"check_modules.py file has been repaired.\")\n",
     "\n",
     "fix_script()\n",
-    "\n",
+    "os.chdir(\"sd-ui-files\")\n",
     "!python ./scripts/check_modules.py  #安装依赖&环境\n",
-    "\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 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",
     "\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",
-    "os.chdir(\"sd-ui-files\")\n",
     "!python ./scripts/check_modules.py --launch-uvicorn"
    ]
   }

easydiffusion/_start.ipynb

@@ -11,15 +11,16 @@
     "import os\n",
     "if not os.path.exists('/content/key/id_rsa'):\n",
     "  !mkdir -p /content/key/id_rsa\n",
-    "  !chmod 600 /content/key/id_rsa\n",
-    "  !ssh-keygen -t rsa -b 4096 -N \"\" -f /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",
     "  from pyngrok import ngrok\n",
     "  ngrok_tunnel = ngrok.connect(9000, \"http\")\n",
     "  from pycloudflared import try_cloudflare\n",
-    "  cloudflare_url = try_cloudflare(9000, verbose=False)\n",
+    "  cloudflare_url = try_cloudflare(9000)\n",
     "  print(ngrok_tunnel)\n",
     "  print(cloudflare_url)\n",
     "  time.sleep(1)\n",
@@ -27,7 +28,6 @@
     "\n",
     "threading.Thread(target=tunnel, daemon=True).start()\n",
     "\n",
-    "os.chdir(\"sd-ui-files\")\n",
     "!python ./scripts/check_modules.py --launch-uvicorn\n",
     "#!python ./scripts//ui/main.py"
    ]