e2tts mlx

E2TTS-MLX: смущающе легко неавторегрессивные с нулевые выстрелы в MLX

Легкая реализация смущающей простой полностью неавторегрессивной модели с нулевым выстрелом с использованием MLX с минимальными зависимостями и эффективными вычислениями на кремниевых условиях Apple.

Быстрый старт

Установить

 # 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 .

Использование

Чтобы использовать предварительно обученную модель для текста в речь:

e2tts ' We must achieve our own salvation. '

Это напишет tts_0.wav в текущий каталог, в который вы можете сыграть.

tts_0.mp4

Чтобы обучить новую модель с настройками по умолчанию:

e2tts

Тренироваться с пользовательскими опциями:

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

Выберите варианты обучения:

• --batch_size : установите размер партии (по умолчанию: 32)
• --n_epoch : Установите количество эпох (по умолчанию: 10)
• --lr : установите скорость обучения (по умолчанию: 2E-4)
• --depth : установите глубину модели (по умолчанию: 8)
• --n_ode : установите количество шагов для отбора проб (по умолчанию: 1)
• --more_ds Параметр: реализует два набора обучения (по умолчанию: 'Josefalbers/LJ-Speech')

Благодарности

Особая благодарность фантастическому коду Lucidrains, который вдохновил этот проект, а также в реализацию Lucasnewman The Vocos, которая сделала это возможным.

Лицензия

Apache License 2.0