Paillat-dev/FABLE
1
0
Fork 0
mirror of https://github.com/Paillat-dev/FABLE.git synced 2026-01-02 01:06:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(speak.py): handle TTS initialization error when gpu is not available

Browse Source
This commit is contained in:
Paillat
2023-05-15 12:32:14 +02:00
parent be20abb833
commit 393600ad94
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
generators/speak.py
View File

@@ -1,7 +1,5 @@
from TTS.api import TTS from TTS.api import TTS
import os
# Running a multi-speaker and multi-lingual model # Running a multi-speaker and multi-lingual model
# List available 🐸TTS models and choose the first one # List available 🐸TTS models and choose the first one
@@ -18,6 +16,9 @@ voices = ["Alexander", "Benjamin", "Amelia", "Katherine"]
# Init TTS # Init TTS
def generate_voice(path, text, speaker="Alexander"): def generate_voice(path, text, speaker="Alexander"):
try:
tts = TTS(model_best_multi, gpu=True) tts = TTS(model_best_multi, gpu=True)
except:
tts = TTS(model_best_multi, gpu=False)
speaker = fakenames[speaker] if speaker in fakenames else speaker speaker = fakenames[speaker] if speaker in fakenames else speaker
tts.tts_to_file(text=text, file_path=path, speaker=speaker, speed=1) tts.tts_to_file(text=text, file_path=path, speaker=speaker, speed=1)