One Click VITS Training

• Treinamento com um clique de um clique

Esta ferramenta permite que você conclua todo o processo de VITs (pré-processamento de dados + sussurro ASR + Texto Pré-processamento + Modificação Config.json + Treinamento, Inferência) Com um clique!

Índice

• Estrutura
• Pré -requisitos
• Instalação
• Prepare_datasets
• Uso
• Inferência
• Mudanças
• Referências

Estrutura

Pré -requisitos

• Um sistema Windows/Linux com um mínimo de RAM de 16GB .
• Uma GPU com pelo menos 12GB de VRAM.
• Python == 3.8
• Anaconda instalada.
• Pytorch instalado.
• CUDA 11.x instalado.
• Zlib DLL instalado.

Comando de instalação pytorch:

pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117

CUDA 11.7 Instale: https://developer.nvidia.com/cuda-11-7-0-download-archive

ZLIB DLL Instalação: https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#install-zlib-windows

Instale o pyopenjtalk manualmente: pip install -U pyopenjtalk --no-build-isolation

---

Instalação

1. Crie um ambiente de Anaconda:

conda create -n one-click-vits python=3.8

2. Ative o ambiente:

conda activate one-click-vits

3. Clone este repositório para sua máquina local:

git clone https://github.com/ORI-Muchim/One-Click-VITS-Training.git

4. Navegue até o diretório clonado:

 cd One-Click-VITS-Training

5. Instale as dependências necessárias:

pip install -r requirements.txt

---

Prepare_datasets

Coloque os arquivos de áudio da seguinte forma.

Os arquivos .mp3 ou .wav estão bem.

 One-Click-VITS-Training
├────datasets
│       ├───speaker0
│       │   ├────1.mp3
│       │   └────1.wav
│       └───speaker1
│       │    ├───1.mp3
│       │    └───1.wav
│       ├integral.py
│       └integral_low.py
│
├────vits
├────inference.py
├────main_low.py
├────main_resume.py
├────main.py
├────Readme.md
└────requirements.txt

Este é apenas um exemplo, e não há problema em adicionar mais alto -falantes.

---

Uso

Para iniciar esta ferramenta, use o seguinte comando, substituindo {idioma}, {model_name} e {sample_rate} com seus respectivos valores ({idioma: ko, ja, en, zh} / {sample_rate: 22050 /44100}):

python main.py {language} {model_name} {sample_rate}

Para aqueles com baixas especificações (VRAM <12 GB), use este código:

python main_low.py {language} {model_name} {sample_rate}

Se a configuração de dados estiver concluída e você quiser retomar o treinamento, insira este código:

python main_resume.py {model_name}

---

Inferência

Depois que o modelo foi treinado, você pode gerar previsões usando o seguinte comando, substituindo {Model_name} e {Model_Step} pelos seus respectivos valores:

python inference.py {model_name} {model_step}

Ou verifique ./vits/inference.ipynb.

Se você deseja alterar o exemplo de texto usado na referência, modifique ./vits/inferencems.py Parte text .

---

Mudanças

No repositório de Cjangcjengh/Vits, fiz algumas modificações no método de limpeza de texto coreano. O outro processo de limpeza é o mesmo, publicando -o no repositório Cjangcjengh, mas o arquivo mais limpo foi modificado usando a biblioteca Tenebo/G2PK2 como o coreano pronunciado.

---

Referências

Para mais informações, consulte os seguintes repositórios:

• JAYWALNUT310/VITS
• Cjangcjengh/vits
• Kyubyong/G2PK
• Tenebo/G2pk2
• Henrymass/Audioslicer