Upload 2 files

sdwui-start-util.ipynb

@@ -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",
@@ -1558,6 +1579,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",
@@ -1705,7 +1727,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.11"
+   "version": "3.10.13"
   }
  },
  "nbformat": 4,

sdwui-start-util.mini.ipynb

@@ -1287,7 +1287,7 @@
    "outputs": [],
    "source": [
     "envInstalled=False\n",
-    "quickStart = True\n",
+    "quickStart = False\n",
     "#安装\n",
     "def install():\n",
     "    print('安装')\n",
@@ -1354,9 +1354,27 @@
     "    if not Path(f'{_install_path}/{_ui_dir_name}/venv/bin/pip').exists():\n",
     "        run('curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py')\n",
     "        run(f'{_install_path}/{_ui_dir_name}/venv/bin/python3 get-pip.py')\n",
-    "\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",
+    "    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",
@@ -1410,7 +1428,6 @@
     "        root_path = _sub_path[i]\n",
     "        if root_path.endswith('/'): root_path = root_path[:-1]\n",
     "        if torch.cuda.device_count() == 2 and not _multi_case:\n",
-    "            print(f'当前同时使用多张显卡进行并行推理提高出图速度')\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",
@@ -1488,6 +1505,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",