feat(colab): add rmvpe

This commit is contained in:
magic-akari 2023-07-12 18:52:09 +08:00
parent f808d8e60b
commit d3e21649bb
No known key found for this signature in database
GPG Key ID: EC005B1159285BDD
1 changed files with 9 additions and 2 deletions

View File

@ -317,9 +317,12 @@
"#@markdown\n",
"%cd /content/so-vits-svc\n",
"\n",
"f0_predictor = \"crepe\" #@param [\"crepe\", \"pm\", \"dio\", \"harvest\"]\n",
"f0_predictor = \"crepe\" #@param [\"crepe\", \"pm\", \"dio\", \"harvest\", \"rmvpe\"]\n",
"use_diff = True #@param {type:\"boolean\"}\n",
"\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",
"diff_param = \"\"\n",
"if use_diff:\n",
" diff_param = \"--use_diff\"\n",
@ -602,7 +605,7 @@
"if auto_predict_f0:\n",
" apf = \" -a \"\n",
"\n",
"f0_predictor = \"crepe\" #@param [\"crepe\", \"pm\", \"dio\", \"harvest\"]\n",
"f0_predictor = \"crepe\" #@param [\"crepe\", \"pm\", \"dio\", \"harvest\", \"rmvpe\"]\n",
"\n",
"enhance = False #@param {type:\"boolean\"}\n",
"ehc = \"\"\n",
@ -618,6 +621,10 @@
"url, output = get_speech_encoder(config_path)\n",
"\n",
"import os\n",
"\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 not os.path.exists(output):\n",
" !curl -L {url} -o {output}\n",
"\n",