auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-05-08 14:28:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Some fixes to the model exporter.

Browse Source
This commit is contained in:
Mateo Cedillo
2023-06-03 18:59:51 -05:00
parent f4b2134107
commit e135c0ff8a
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
notebooks/piper_model_exporter.ipynb
View File

@@ -5,7 +5,7 @@
"colab": { "colab": {
"provenance": [], "provenance": [],
"gpuType": "T4", "gpuType": "T4",
"authorship_tag": "ABX9TyMkrP91PCGT541xgO1Y2fSe", "authorship_tag": "ABX9TyOyCiCt4X4oZWyqzQ2fAafX",
"include_colab_link": true "include_colab_link": true
}, },
"kernelspec": { "kernelspec": {
@@ -14,9 +14,7 @@
}, },
"language_info": { "language_info": {
"name": "python" "name": "python"
}, }
"accelerator": "GPU",
"gpuClass": "standard"
}, },
"cells": [ "cells": [
{ {
@@ -70,14 +68,14 @@
"import os\n", "import os\n",
"#@markdown #### Download:\n", "#@markdown #### Download:\n",
"#@markdown **Google Drive model ID:**\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", "#@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", "#@markdown ---\n",
"\n", "\n",
"#@markdown #### Creation process:\n", "#@markdown #### Creation process:\n",
"language = \"en-us\" #@param [\"en-us\", \"es\", \"fr\"]\n", "language = \"es\" #@param [\"en-us\", \"es\", \"fr\"]\n",
"voice_name = \"Joe\" #@param {type:\"string\"}\n", "voice_name = \"Dabe\" #@param {type:\"string\"}\n",
"voice_name = voice_name.lower()\n", "voice_name = voice_name.lower()\n",
"quality = \"medium\" #@param [\"high\", \"low\", \"medium\", \"x-low\"]\n", "quality = \"medium\" #@param [\"high\", \"low\", \"medium\", \"x-low\"]\n",
"export_voice_name = f\"{language}-{voice_name}-{quality}\"\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", "!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", "!python -m piper_train.export_onnx \"/content/project/model.ckpt\" \"{export_voice_path}/{export_voice_name}.onnx\"\n",
"print(\"compressing...\")\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": { "metadata": {
"cellView": "form", "cellView": "form",