diff --git a/notebooks/so-vits-svc-fork-4.0.ipynb b/notebooks/so-vits-svc-fork-4.0.ipynb index ba7b6765..91471c1b 100644 --- a/notebooks/so-vits-svc-fork-4.0.ipynb +++ b/notebooks/so-vits-svc-fork-4.0.ipynb @@ -1,14 +1,13 @@ { "cells": [ { + "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "## Before training\n", "\n", - "This program saves the last 3 generations of models to Google Drive. Since 1 generation of models is >1GB, you should have at least 3GB of free space in Google Drive. If you do not have such free space, it is recommended to create another Google Account.\n", - "\n", - "Training requires >10GB VRAM. (T4 should be enough) Inference does not require such a lot of VRAM." + "This program saves the last 3 generations of models to Google Drive. Since 1 generation of models is >1GB, you should have at least 3GB of free space in Google Drive. If you do not have such free space, it is recommended to create another Google Account." ] }, { @@ -18,16 +17,6 @@ "## Installation" ] }, - { - "cell_type": "code", - "execution_count": null, - "metadata": {}, - "outputs": [], - "source": [ - "#@title Check GPU\n", - "!nvidia-smi" - ] - }, { "cell_type": "code", "execution_count": null, @@ -52,9 +41,13 @@ "\n", "#@markdown Branch (for development)\n", "BRANCH = \"none\" #@param {\"type\": \"string\"}\n", + "TPU = True #@param {\"type\": \"boolean\"}\n", + "if TPU:\n", + " %pip install cloud-tpu-client==0.10 torch==2.0.0 torchvision==0.15.1 https://storage.googleapis.com/tpu-pytorch/wheels/colab/torch_xla-2.0-cp39-cp39-linux_x86_64.whl\n", "if BRANCH == \"none\":\n", " %pip install -U so-vits-svc-fork\n", "else:\n", + " %pip uninstall -y so-vits-svc-fork\n", " %pip install -U git+https://github.com/34j/so-vits-svc-fork.git@{BRANCH}" ] }, @@ -94,7 +87,7 @@ "#@title Copy your dataset\n", "#@markdown **We assume that your dataset is in your Google Drive's `so-vits-svc-fork/dataset/(speaker_name)` directory.**\n", "DATASET_NAME = \"kiritan\" #@param {type: \"string\"}\n", - "!cp -R /content/drive/MyDrive/so-vits-svc-fork/dataset/{DATASET_NAME}/ -t \"dataset_raw/\"" + "!cp -R /content/drive/MyDrive/so-vits-svc-fork/dataset_raw_raw/{DATASET_NAME}/ -t \"dataset_raw/\"" ] }, { @@ -119,16 +112,7 @@ "outputs": [], "source": [ "#@title Automatic preprocessing\n", - "!svc pre-resample" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": {}, - "outputs": [], - "source": [ - "!svc pre-config" + "!svc pre-resample -o \"drive/MyDrive/so-vits-svc-fork/dataset_raw\"" ] }, { @@ -137,8 +121,7 @@ "metadata": {}, "outputs": [], "source": [ - "#@title Copy configs file\n", - "!cp configs/44k/config.json drive/MyDrive/so-vits-svc-fork" + "!svc pre-config -i \"drive/MyDrive/so-vits-svc-fork/dataset_raw\" -o \"drive/MyDrive/so-vits-svc-fork/configs/44k\"" ] }, { @@ -148,7 +131,7 @@ "outputs": [], "source": [ "F0_METHOD = \"dio\" #@param [\"crepe\", \"crepe-tiny\", \"parselmouth\", \"dio\", \"harvest\"]\n", - "!svc pre-hubert -fm {F0_METHOD}" + "!svc pre-hubert -fm {F0_METHOD} -i \"drive/MyDrive/so-vits-svc-fork/dataset\" -c \"drive/MyDrive/so-vits-svc-fork/configs/44k\"" ] }, { @@ -160,7 +143,7 @@ "#@title Train\n", "%load_ext tensorboard\n", "%tensorboard --logdir drive/MyDrive/so-vits-svc-fork/logs/44k\n", - "!svc train --model-path drive/MyDrive/so-vits-svc-fork/logs/44k" + "!svc train -m \"drive/MyDrive/so-vits-svc-fork/logs/44k\" -c \"drive/MyDrive/so-vits-svc-fork/configs/44k\"" ] }, {