aljaz/nostrdvm
1
0
Fork 0
mirror of https://github.com/aljazceru/nostrdvm.git synced 2025-12-24 09:24:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

better handling of dvms venvs (again with live feedback)

Browse Source
This commit is contained in:
Believethehype
2024-06-26 10:55:17 +02:00
parent 000a0fc983
commit cadacb59f0
3 changed files with 87 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
nostr_dvm/tasks/texttospeech.py
View File

@@ -106,9 +106,11 @@ class TextToSpeech(DVMTaskInterface):
from TTS.api import TTS
options = self.set_options(request_form)
device = "cuda" if torch.cuda.is_available() else "cpu"
# else "mps" if torch.backends.mps.is_available() \
#else "mps" if torch.backends.mps.is_available()
print(device)
print(TTS().list_models().list_tts_models())
print(TTS().list_models())
try:
# model = "tts_models/deu/fairseq/vits"
# model = "tts_models/multilingual/multi-dataset/your_tts"