mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Ability to run the Coqui TTS extension on the CPU (#5365)

Browse source
This commit is contained in:
Hubert Kasperek 2024-01-27 21:15:34 +01:00 • committed by GitHub
parent 828be63f2c
commit 69622930c7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
extensions/coqui_tts/script.py
View file

@ -3,6 +3,7 @@ import json
import os import os
import random import random
import time import time
import torch
from pathlib import Path from pathlib import Path
import gradio as gr import gradio as gr
@ -23,7 +24,7 @@ params = {
"voice": "female_01.wav", "voice": "female_01.wav",
"language": "English", "language": "English",
"model_name": "tts_models/multilingual/multi-dataset/xtts_v2", "model_name": "tts_models/multilingual/multi-dataset/xtts_v2",
"device": "cuda" "device": "cuda" if torch.cuda.is_available() else "cpu"
} }
this_dir = str(Path(__file__).parent.resolve()) this_dir = str(Path(__file__).parent.resolve())