Some fixes to the model exporter.

This commit is contained in:
Mateo Cedillo
2023-06-03 18:59:51 -05:00
parent f4b2134107
commit e135c0ff8a

View File

@@ -5,7 +5,7 @@
"colab": {
"provenance": [],
"gpuType": "T4",
"authorship_tag": "ABX9TyMkrP91PCGT541xgO1Y2fSe",
"authorship_tag": "ABX9TyOyCiCt4X4oZWyqzQ2fAafX",
"include_colab_link": true
},
"kernelspec": {
@@ -14,9 +14,7 @@
},
"language_info": {
"name": "python"
},
"accelerator": "GPU",
"gpuClass": "standard"
}
},
"cells": [
{
@@ -70,14 +68,14 @@
"import os\n",
"#@markdown #### Download:\n",
"#@markdown **Google Drive model ID:**\n",
"model_id = \"1KTwFwDy17RdPDT2qDJsyty__rt3KviZZ\" #@param {type:\"string\"}\n",
"model_id = \"\" #@param {type:\"string\"}\n",
"#@markdown **Config.json Google Drive ID**\n",
"config_id = \"1-hEqdUetfaqrSYZF8hB75Y-9ZR0S--qa\" #@param {type:\"string\"}\n",
"config_id = \"1-7qJ-Aw8OmqKgxxG8o_onU_Rwn0xfIOj\" #@param {type:\"string\"}\n",
"#@markdown ---\n",
"\n",
"#@markdown #### Creation process:\n",
"language = \"en-us\" #@param [\"en-us\", \"es\", \"fr\"]\n",
"voice_name = \"Joe\" #@param {type:\"string\"}\n",
"language = \"es\" #@param [\"en-us\", \"es\", \"fr\"]\n",
"voice_name = \"Dabe\" #@param {type:\"string\"}\n",
"voice_name = voice_name.lower()\n",
"quality = \"medium\" #@param [\"high\", \"low\", \"medium\", \"x-low\"]\n",
"export_voice_name = f\"{language}-{voice_name}-{quality}\"\n",
@@ -91,7 +89,8 @@
"!gdown \"{config_id}\" -O \"{export_voice_path}/{export_voice_name}.onnx.json\"\n",
"!python -m piper_train.export_onnx \"/content/project/model.ckpt\" \"{export_voice_path}/{export_voice_name}.onnx\"\n",
"print(\"compressing...\")\n",
"!tar -czvf \"{packages_path}/voice-{export_voice_name}.tar.gz\" \"{export_voice_path}\""
"%cd /content/project\n",
"!tar -czvf \"{packages_path}/voice-{export_voice_name}.tar.gz\" \"voice-{export_voice_name}\""
],
"metadata": {
"cellView": "form",