docs(notebook): add cluster training and inference (#84)

voicepaw · Mar 23, 2023 · 9ffb621 · 9ffb621
1 parent 2018591
commit 9ffb621
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/notebooks/so-vits-svc-fork-4.0.ipynb b/notebooks/so-vits-svc-fork-4.0.ipynb
@@ -170,6 +170,23 @@
     "!svc train --model-path drive/MyDrive/so-vits-svc-fork/logs/44k"
    ]
   },
+  {
+   "attachments": {},
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Training Cluster model"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!svc train-cluster --output-path drive/MyDrive/so-vits-svc-fork/logs/44k"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -202,6 +219,17 @@
     "display(Audio(f\"{NAME}.out.wav\", autoplay=True))"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "##@title Use trained model (with cluster)\n",
+    "!svc infer-cluster {NAME}.wav -r 0.1 -m drive/MyDrive/so-vits-svc-fork/logs/44k/ -c drive/MyDrive/so-vits-svc-fork/logs/44k/config.json -k drive/MyDrive/so-vits-svc-fork/logs/44k/kmeans.pt\n",
+    "display(Audio(f\"{NAME}.out.wav\", autoplay=True))"
+   ]
+  },
   {
    "attachments": {},
    "cell_type": "markdown",