Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 10:35:10 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

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
WARNING! Although there is a key with this ID in the database it does not verify this commit! This commit is SUSPICIOUS.
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show stats 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())