Upload 2 files

easydiffusion/_main.ipynb

@@ -77,10 +77,9 @@
     "  gen_key(ssh_path)\n",
     "\n",
     "os.chdir(\"sd-ui-files\")\n",
-    "main_dir = os.environ['MAIN_DIR'] = os.getcwd() if 'MAIN_DIR' not in os.environ else os.environ['MAIN_DIR']\n",
     "\n",
     "!python ./scripts/check_modules.py  #安装依赖&环境\n",
-    "!pip install --upgrade pytorch-lightning\n",
+    "!pip install --upgrade pytorch-lightning pycloudflared\n",
     "!apt-get install aria2\n",
     "#!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M -d {current_dir}/models/stable-diffusion/ -o chilloutmix_NiPrunedFp32Fix.safetensors https://huggingface.co/naonovn/chilloutmix_NiPrunedFp32Fix/resolve/main/chilloutmix_NiPrunedFp32Fix.safetensors\n",
     "!aria2c --console-log-level=error -c -x 16 -s 16 -k 1M -d {current_dir}/models/stable-diffusion/ -o chilloutmix_NiPrunedFp32Fix-chonghui.safetensors https://huggingface.co/spaces/weo1101/111/resolve/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting.safetensors\n",

easydiffusion/_start.ipynb

@@ -11,16 +11,14 @@
     "import time\n",
     "import os\n",
     "#/content/esay\n",
-    "current_dir = os.environ['CURRENT_DIR']\n",
-    "main_dir = os.environ['MAIN_DIR']\n",
-    "%cd {main_dir}\n",
+    "%cd /content/esay/sd-ui-files\n",
     "use_remote = True # @param {type:\"boolean\"}\n",
     "use_cloudflare = True # @param {type:\"boolean\"}\n",
     "\n",
     "def start_tunnle():\n",
     "  open(\"/content/output.log\", \"w\").close()\n",
     "  # 循环检测文件内容，直到存在\"http://127.0.0.1:9090\"。\n",
-    "  while not \"Warning: Could not find TensorRT\" in open(\"/content/output.log\", \"r\").read():\n",
+    "  while not \"http://localhost:9000/\" in open(\"/content/output.log\", \"r\").read():\n",
     "    time.sleep(1)\n",
     "  clear_output()  \n",
     "  if ngrok_token:\n",
@@ -41,13 +39,13 @@
     "\n",
     "  if use_remote:\n",
     "    try:\n",
-    "         !ssh -R 80:127.0.0.1:9000 -o StrictHostKeyChecking=no -i {current_dir}/id_rsa remote.moe 2>&1 | tee -a /content/tunnel.log\n",
+    "         !ssh -R 80:127.0.0.1:9000 -o StrictHostKeyChecking=no -i /content/id_rsa remote.moe 2>&1 | tee -a /content/tunnel.log\n",
     "    except Exception as e:\n",
     "        print(\"cloudflare 连接失败：\", e)\n",
     "\n",
     "threading.Thread(target=start_tunnle, daemon=True).start()\n",
     "\n",
-    "!python {current_dir}/sd-ui-files/scripts/check_modules.py --launch-uvicorn 2>&1 | tee /content/output.log\n",
+    "!python ./scripts/check_modules.py --launch-uvicorn 2>&1 | tee /content/output.log\n",
     "#!python ./ui/main.py"
    ]
   }