leon-ai/leon
1
1
Fork 0
mirror of https://github.com/leon-ai/leon.git synced 2024-11-30 19:07:39 +03:00
Code Issues Projects Releases Wiki Activity

feat(python tcp server): map speech synthesis hardware device choice to settings

Browse Source
This commit is contained in:
louistiti 2024-06-30 16:54:59 +08:00
parent a05f34422e
commit 8eeff3a01c
No known key found for this signature in database
GPG Key ID: 92CD6A2E497E1669
3 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tcp_server/settings.json
View File

@ -4,7 +4,8 @@
"device": "auto"
},
"tts": {
"model_file_name": "EN-Leon-V1_1-G_600000.pth"
"model_file_name": "EN-Leon-V1_1-G_600000.pth",
"device": "auto"
},
"wake_word": {}
}

2
tcp_server/src/lib/tcp_server.py
View File

@ -58,7 +58,7 @@ class TCPServer:
return
self.tts = TTS(language='EN',
device='cpu',
device=get_settings('tts')['device'],
config_path=TTS_MODEL_CONFIG_PATH,
ckpt_path=TTS_MODEL_PATH
)

1
tcp_server/src/lib/tts/api.py
View File

@ -18,6 +18,7 @@ from ..utils import is_macos
class TTS(nn.Module):
def __init__(self,
language,
# auto, cpu, cuda, mps
device='auto',
use_hf=True,
config_path=None,