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

Force coqui tts agree

Browse Source
This commit is contained in:
Paillat
2024-03-01 12:49:09 +01:00
parent 2dbd707817
commit f1db6b27ce
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/engines/TTSEngine/CoquiTTSEngine.py
View File

@@ -99,9 +99,11 @@ class CoquiTTSEngine(BaseTTSEngine):
self.to_force_duration = options[2] self.to_force_duration = options[2]
self.duration = options[3] self.duration = options[3]
os.environ["COQUI_TOS_AGREED"] = "1" os.environ["COQUI_TOS_AGREED"] = options[4]
try:
self.tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2") self.tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2")
except:
raise Exception("An error occured when loading thr TTS model. Make sure that you have agreed to the TOS in the TTSEngine tab.")
device = "cuda" if torch.cuda.is_available() else "cpu" device = "cuda" if torch.cuda.is_available() else "cpu"
self.tts.to(device) self.tts.to(device)
@@ -159,4 +161,11 @@ class CoquiTTSEngine(BaseTTSEngine):
options.append(duration_checkbox) options.append(duration_checkbox)
options.append(duration) options.append(duration)
options.append(gr.Checkbox(
label="I agree to the Coqui public mode license",
info="You must agree to the Coqui TTS terms of service to use this engine: https://coqui.ai/cpml",
value=False,
show_label=True,
))
return options return options