e2tts mlx

E2TTS-MLX：MLX中令人尴尬地轻松完全非自动回旋零射击TTS

使用MLX轻松实现了令人尴尬的完全非自动回归零击中TTS模型的实现，对苹果硅的依赖性最小和有效的计算。

快速开始

安装

 # 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的Vocos实施，这使得这一目标成为可能。

执照

Apache许可证2.0