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