e2tts mlx

E2TTS-MLX: TT Zero-shot yang sangat mudah dan memalukan sepenuhnya di MLX di MLX

Implementasi ringan dari model TTS Zero-Shot yang sangat mudah memalukan menggunakan MLX, dengan dependensi minimal dan perhitungan yang efisien pada silikon apel.

Awal yang cepat

Memasang

 # Quick install (note: PyPI version may not always be up to date)
pip install e2tts-mlx

# For the latest version, you can install directly from the repository:
# git clone https://github.com/JosefAlbers/e2tts-mlx.git
# cd e2tts-mlx
# pip install -e .

Penggunaan

Untuk menggunakan model pra-terlatih untuk teks-ke-speech:

e2tts ' We must achieve our own salvation. '

Ini akan menulis tts_0.wav ke direktori saat ini, yang kemudian dapat Anda mainkan.

tts_0.mp4

Untuk melatih model baru dengan pengaturan default:

e2tts

Untuk berlatih dengan opsi khusus:

e2tts --batch_size=16 --n_epoch=100 --lr=1e-4 --depth=8 --n_ode=32

Pilih Opsi Pelatihan:

• --batch_size : Atur ukuran batch (default: 32)
• --n_epoch : Atur jumlah zaman (default: 10)
• --lr : atur tingkat pembelajaran (default: 2e-4)
• --depth : Atur kedalaman model (default: 8)
• --n_ode : Setel jumlah langkah untuk pengambilan sampel (default: 1)
• Parameter --more_ds : mengimplementasikan pelatihan dua set (default: 'josefalbers/lj-speech')

Ucapan Terima Kasih

Terima kasih khusus kepada kode fantastis Lucidrains yang menginspirasi proyek ini, dan kepada Lucasnewman adalah implementasi VOCOS yang memungkinkan hal ini.

Lisensi

Lisensi Apache 2.0