Update colab

colab.ipynb

@@ -4,7 +4,7 @@
       "cell_type": "markdown",
       "metadata": {},
       "source": [
-        "# Style-Bert-VITS2 (ver 1.3.1) のGoogle Colabでの学習\n",
+        "# Style-Bert-VITS2 (ver 2.0) のGoogle Colabでの学習\n",
         "\n",
         "Google Colab上でStyle-Bert-VITS2の学習を行うことができます。\n",
         "\n",
@@ -223,7 +223,10 @@
         "normalize = False\n",
         "\n",
         "# 音声ファイルの開始・終了にある無音区間を削除するかどうか\n",
-        "trim = False"
+        "trim = False\n",
+        "\n",
+        "# JP-Extra （日本語特化版）を使うかどうか。日本語の能力が向上する代わりに英語と中国語は使えなくなります。\n",
+        "use_jp_extra = True"
       ]
     },
     {
@@ -256,6 +259,11 @@
         "    num_processes=2,\n",
         "    normalize=normalize,\n",
         "    trim=trim,\n",
+        "    freeze_EN_bert=False,\n",
+        "    freeze_JP_bert=False,\n",
+        "    freeze_ZH_bert=False,\n",
+        "    freeze_style=False,\n",
+        "    use_jp_extra=use_jp_extra\n",
         ")"
       ]
     },
@@ -297,8 +305,26 @@
         "    yml_data = yaml.safe_load(f)\n",
         "yml_data[\"model_name\"] = model_name\n",
         "with open(\"config.yml\", \"w\", encoding=\"utf-8\") as f:\n",
-        "    yaml.dump(yml_data, f, allow_unicode=True)\n",
+        "    yaml.dump(yml_data, f, allow_unicode=True)"
-        "\n",
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# 日本語特化版を「使う」場合\n",
+        "!python train_ms_jp_extra.py --config {config_path} --model {dataset_path} --assets_root {assets_root}"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# 日本語特化版を「使わない」場合\n",
         "!python train_ms.py --config {config_path} --model {dataset_path} --assets_root {assets_root}"
       ]
     },