feat(colab): add fcpe

This commit is contained in:
magic-akari 2023-07-26 14:43:24 +08:00
parent bb610c5805
commit a5dac31f43
No known key found for this signature in database
GPG Key ID: EC005B1159285BDD

View File

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