From a5dac31f43ce9e39e880b1267d70a5035f5bac74 Mon Sep 17 00:00:00 2001
From: magic-akari <akari.ccino@gmail.com>
Date: Wed, 26 Jul 2023 14:43:24 +0800
Subject: [PATCH] feat(colab): add `fcpe`

---
 sovits4_for_colab.ipynb | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/sovits4_for_colab.ipynb b/sovits4_for_colab.ipynb
index 4f77592..0d47a4b 100644
--- a/sovits4_for_colab.ipynb
+++ b/sovits4_for_colab.ipynb
@@ -316,13 +316,17 @@
     "#@markdown\n",
     "%cd /content/so-vits-svc\n",
     "\n",
-    "f0_predictor = \"crepe\" #@param [\"crepe\", \"pm\", \"dio\", \"harvest\", \"rmvpe\"]\n",
+    "f0_predictor = \"crepe\" #@param [\"crepe\", \"pm\", \"dio\", \"harvest\", \"rmvpe\", \"fcpe\"]\n",
     "use_diff = True #@param {type:\"boolean\"}\n",
     "\n",
     "import os\n",
     "if f0_predictor == \"rmvpe\" and not os.path.exists(\"./pretrain/rmvpe.pt\"):\n",
     "  !curl -L https://huggingface.co/datasets/ylzz1997/rmvpe_pretrain_model/resolve/main/rmvpe.pt -o pretrain/rmvpe.pt\n",
     "\n",
+    "if f0_predictor == \"fcpe\" and not os.path.exists(\"./pretrain/fcpe.pt\"):\n",
+    "  !curl -L https://huggingface.co/datasets/ylzz1997/rmvpe_pretrain_model/resolve/main/fcpe.pt -o pretrain/fcpe.pt\n",
+    "\n",
+    "\n",
     "diff_param = \"\"\n",
     "if use_diff:\n",
     "  diff_param = \"--use_diff\"\n",
@@ -623,7 +627,7 @@
     "if auto_predict_f0:\n",
     "  apf = \" -a \"\n",
     "\n",
-    "f0_predictor = \"crepe\" #@param [\"crepe\", \"pm\", \"dio\", \"harvest\", \"rmvpe\"]\n",
+    "f0_predictor = \"crepe\" #@param [\"crepe\", \"pm\", \"dio\", \"harvest\", \"rmvpe\", \"fcpe\"]\n",
     "\n",
     "enhance = False  #@param {type:\"boolean\"}\n",
     "ehc = \"\"\n",
@@ -643,6 +647,9 @@
     "if f0_predictor == \"rmvpe\" and not os.path.exists(\"./pretrain/rmvpe.pt\"):\n",
     "  !curl -L https://huggingface.co/datasets/ylzz1997/rmvpe_pretrain_model/resolve/main/rmvpe.pt -o pretrain/rmvpe.pt\n",
     "\n",
+    "if f0_predictor == \"fcpe\" and not os.path.exists(\"./pretrain/fcpe.pt\"):\n",
+    "  !curl -L https://huggingface.co/datasets/ylzz1997/rmvpe_pretrain_model/resolve/main/fcpe.pt -o pretrain/fcpe.pt\n",
+    "\n",
     "if not os.path.exists(output):\n",
     "  !curl -L {url} -o {output}\n",
     "\n",