From c6acb9e8ce5ad2bd41558a6f4b9fb2fb24a0d4cf Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Mon, 5 Feb 2024 16:04:39 +0900
Subject: [PATCH] Add colab merge

---
 colab.ipynb    | 24 ++++++++++++++++++++----
 webui_merge.py |  5 ++++-
 2 files changed, 24 insertions(+), 5 deletions(-)

diff --git a/colab.ipynb b/colab.ipynb
index fa1134a..30cf3cd 100644
--- a/colab.ipynb
+++ b/colab.ipynb
@@ -209,6 +209,9 @@
         "# 上でつけたフォルダの名前`Data/{model_name}/`\n",
         "model_name = \"your_model_name\"\n",
         "\n",
+        "# JP-Extra （日本語特化版）を使うかどうか。日本語の能力が向上する代わりに英語と中国語は使えなくなります。\n",
+        "use_jp_extra = True\n",
+        "\n",
         "# 学習のバッチサイズ。VRAMのはみ出具合に応じて調整してください。\n",
         "batch_size = 4\n",
         "\n",
@@ -223,10 +226,7 @@
         "normalize = False\n",
         "\n",
         "# 音声ファイルの開始・終了にある無音区間を削除するかどうか\n",
-        "trim = False\n",
-        "\n",
-        "# JP-Extra （日本語特化版）を使うかどうか。日本語の能力が向上する代わりに英語と中国語は使えなくなります。\n",
-        "use_jp_extra = True"
+        "trim = False"
       ]
     },
     {
@@ -359,6 +359,22 @@
       "source": [
         "!python webui_style_vectors.py --share"
       ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {},
+      "source": [
+        "## 6. マージ"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "!python webui_merge.py --share"
+      ]
     }
   ],
   "metadata": {
diff --git a/webui_merge.py b/webui_merge.py
index 6c8b8c1..ed0f37d 100644
--- a/webui_merge.py
+++ b/webui_merge.py
@@ -430,6 +430,9 @@ parser.add_argument(
     default=False,
     help="Do not launch app automatically",
 )
+parser.add_argument("--share", action="store_true", default=False)
 args = parser.parse_args()
 
-app.launch(inbrowser=not args.no_autolaunch, server_name=args.server_name)
+app.launch(
+    inbrowser=not args.no_autolaunch, server_name=args.server_name, share=args.share
+)