{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "private_outputs": true,
      "provenance": [],
      "gpuType": "T4"
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU"
  },
  "cells": [
    {
      "cell_type": "code",
      "source": [
        "#@title Check GPU\n",
        "!nvidia-smi"
      ],
      "metadata": {
        "id": "UYsjRYl6IGdY"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Mount Google Drive\n",
        "from google.colab import drive\n",
        "drive.mount('/content/drive')"
      ],
      "metadata": {
        "id": "XDF6CLukIMBY"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Install dependencies\n",
        "#@markdown pip may fail to resolve dependencies and raise ERROR, but it can be ignored.\n",
        "!python -m pip install -U pip wheel\n",
        "%pip install -U ipython\n",
        "\n",
        "#@markdown Branch (for development)\n",
        "BRANCH = \"none\" #@param {\"type\": \"string\"}\n",
        "if BRANCH == \"none\":\n",
        "    %pip install -U so-vits-svc-fork\n",
        "else:\n",
        "    %pip install -U git+https://github.com/34j/so-vits-svc-fork.git@{BRANCH}"
      ],
      "metadata": {
        "id": "i_nqeMhAITlQ"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Make dataset directory\n",
        "!mkdir -p \"dataset_raw\""
      ],
      "metadata": {
        "id": "A4mLrAx1IfXJ"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#!rm -r \"dataset_raw\"\n",
        "#!rm -r \"dataset/44k\""
      ],
      "metadata": {
        "id": "XP7DwtUJIh-Q"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Copy your dataset\n",
        "#@markdown Hãy tạo thư mục trên drive và upload các file âm thanh đã được split 10s\n",
        "import shutil\n",
        "# shutil.rmtree(\"/content/dataset_raw/dataset\", ignore_errors=False, onerror=None)\n",
        "DATASET_NAME = \"hana-hoang\" #@param {type: \"string\"}\n",
        "#@markdown Nơi lưu trữ dataset\n",
        "DATASET_LOCATION = \"huggingface\" # @param [\"huggingface\", \"drive\"]\n",
        "if DATASET_LOCATION == \"huggingface\":\n",
        "    !git clone https://huggingface.co/vunhucuongit/sach-noi-model\n",
        "    DATASET_PATH = \"/content/sach-noi-model/hana-hoang/dataset\"\n",
        "    shutil.copytree(DATASET_PATH, \"/content/dataset_raw/dataset/\")\n",
        "else:\n",
        "    DATASET_PATH = f\"/content/drive/MyDrive/TTS/{DATASET_NAME}/\"\n",
        "\n"
      ],
      "metadata": {
        "id": "XAgGNnLzIkpm"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "import shutil\n",
        "shutil.copytree(\"/content/sach-noi-model/hana-hoang/logs/44k\", \"/content/drive/MyDrive/TTS/hana-hoang\")"
      ],
      "metadata": {
        "id": "TgqgcJw-hlAP"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "import shutil\n",
        "shutil.rmtree(\"/content/sach-noi-model\", ignore_errors=False, onerror=None)"
      ],
      "metadata": {
        "id": "ki8j8Fh3i8s0"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Automatic preprocessing\n",
        "!svc pre-resample"
      ],
      "metadata": {
        "id": "qd_DTQkzIoPW"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "!svc pre-config"
      ],
      "metadata": {
        "id": "wiySQuc_Ip4m"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Copy configs file\n",
        "!cp configs/44k/config.json drive/MyDrive/TTS/hana-hoang"
      ],
      "metadata": {
        "id": "YVseyFdTIs6e"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "F0_METHOD = \"dio\" #@param [\"crepe\", \"crepe-tiny\", \"parselmouth\", \"dio\", \"harvest\"]\n",
        "!svc pre-hubert -fm {F0_METHOD}"
      ],
      "metadata": {
        "id": "yHuKImneIuc_"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Train\n",
        "# %load_ext tensorboard\n",
        "# %tensorboard --logdir drive/MyDrive/TTS/hana-hoang/logs/44k\n",
        "!svc train --model-path /content/drive/MyDrive/TTS/hana-hoang"
      ],
      "metadata": {
        "id": "HC1vG1GAIwUG"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}