Upload sdwui-start-util.dev.ipynb

sdwui-start-util.dev.ipynb

@@ -115,6 +115,7 @@
     "    \n",
     "show_shell_info = locals().get('hidden_console_info') or globals().get('hidden_console_info')\n",
     "if show_shell_info is None: show_shell_info = False\n",
+    "else: show_shell_info = not show_shell_info\n",
     "\n",
     "_multi_case = locals().get('multi_case') or globals().get('multi_case') or False\n",
     "    \n",
@@ -284,8 +285,6 @@
     "os.environ['google_drive'] = google_drive\n",
     "os.environ['input_path'] = _input_path\n",
     "\n",
-    "space_string = ' \\n\\r\\t\\'\\\",'\n",
-    "\n",
     "def replace_path(input_str:str):\n",
     "    if not input_str: return ''\n",
     "    for key in _config_args:\n",
@@ -301,9 +300,9 @@
     "    .replace('$output_path',_output_path)\\\n",
     "    .replace('{output_path}',_output_path)\\\n",
     "    .replace('{sdwui}','stable-diffusion-webui')\\\n",
-    "    .replace('{wui}',\"webui\")\\\n",
-    "    .strip()\n",
+    "    .replace('{wui}',\"webui\")\n",
     "\n",
+    "space_string = ' \\n\\r\\t\\'\\\",'\n",
     "\n",
     "def config_reader(conf:str):\n",
     "    conf = conf or \"\"\n",
@@ -501,7 +500,7 @@
     "    if 'frp_ssl_dir' in locals() or 'frp_ssl_dir' in globals():\n",
     "        frpcSSLFFlies = frpcSSLFFlies + config_reader(locals().get('frp_ssl_dir') or globals().get('frp_ssl_dir'))\n",
     "    # frpc 文件目录 如果目录不存在，会自动下载，也可以在数据集搜索 viyiviyi/utils 添加\n",
-    "    frpcExePath = os.path.join(_input_path,'utils-tools/frpc_v0.51.0')\n",
+    "    frpcExePath = os.path.join(_input_path,'utils-tools/frpc')\n",
     "    # 其他需要加载的webui启动参数 写到【参数列表】这个配置去\n",
     "    otherArgs = '--xformers'\n",
     "    if 'sd_start_args' in locals() or 'sd_start_args' in globals():\n",
@@ -1097,8 +1096,9 @@
     "    folder_list = []  \n",
     "    for item in os.listdir(directory):  \n",
     "        if os.path.isdir(os.path.join(directory, item)):  \n",
-    "            folder_list.append(item)  \n",
-    "    return folder_list \n",
+    "            folder_list.append(item) \n",
+    "    return folder_list\n",
+    "\n",
     "def read_text_file(file_path:str):\n",
     "    if not Path(file_path).exists(): return ''\n",
     "    with open(file_path,\"r\") as f:\n",
@@ -1346,7 +1346,7 @@
    "outputs": [],
    "source": [
     "envInstalled=False\n",
-    "quickStart = True\n",
+    "quickStart = False\n",
     "#安装\n",
     "def install():\n",
     "    print('安装')\n",
@@ -1428,7 +1428,28 @@
     "\n",
     "    get_ipython().system(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -V''')\n",
     "    get_ipython().system(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -m pip -V''')\n",
-    "\n",
+    "    \n",
+    "    if not quickStart:\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -V''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -m pip -V''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python -m pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cu121''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python -m pip install xformers==0.0.27''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python -m pip install -r requirements_versions.txt''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python -m pip install spandrel==0.3.0''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python -m pip install opencv-python''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python -m pip install pydantic==1.10.15''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python -m pip install transformers -U''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python -m pip install llama-cpp-python''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python -m pip install pytorch_lightning==2.3.3 torchsde==0.2.6 spandrel==0.3.4''')\n",
+    "        \n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -m pip install open-clip-torch -U''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -m pip install protobuf==4.25.8''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -m pip install blendmodes==2022''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -m pip install Pillow==9.5.0''')\n",
+    "        run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -m pip install basicsr''')\n",
+    "        \n",
+    "    get_ipython().system(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -m pip install basicsr''')\n",
+    "    \n",
     "    envInstalled = True\n",
     "    print('安装需要的python环境 完成')\n",
     "      \n",
@@ -1474,13 +1495,20 @@
     "    if i>0:\n",
     "        print(f'使用第{i+1}张显卡启动第{i+1}个服务，通过frpc或nrgok地址后加{_sub_path[i]}进行访问')\n",
     "\n",
+    "    current_thread = threading.current_thread()\n",
+    "    current_thread.name = f\"solo_webui__{i}\"\n",
+    "    \n",
     "    restart_times = 0\n",
     "    last_restart_time = time.time()\n",
     "    while _runing:\n",
     "        os.chdir(f'{_install_path}/{_ui_dir_name}')\n",
     "        root_path = _sub_path[i]\n",
     "        if root_path.endswith('/'): root_path = root_path[:-1]\n",
-    "        get_ipython().system(f'''venv/bin/python3 launch.py --device-id={i} --port {str(_server_port+1+i)} --subpath={_sub_path[i]}''')\n",
+    "        if torch.cuda.device_count() == 2 and not _multi_case:\n",
+    "            os.environ['CUDA_VISIBLE_DEVICES']='0,1'\n",
+    "            get_ipython().system(f'''venv/bin/python3 launch.py --port {str(_server_port+1+i)} --subpath={_sub_path[i]}''')\n",
+    "        else:    \n",
+    "            get_ipython().system(f'''venv/bin/python3 launch.py --device-id={i} --port {str(_server_port+1+i)} --subpath={_sub_path[i]}''')\n",
     "        print('10秒后重启服务')\n",
     "        if time.time() - last_restart_time < 60:\n",
     "            restart_times = restart_times + 1\n",
@@ -1554,6 +1582,7 @@
     "    proxy_thread.start()\n",
     "    if isInstall is False or _reLoad: \n",
     "        print('安装运行环境')\n",
+    "        os.environ['MPLBACKEND'] = 'Agg'\n",
     "        install()\n",
     "        link_dir()\n",
     "        init_huggingface()\n",