highperfocused/nostrdvm
1
0
Fork 0
mirror of https://github.com/believethehype/nostrdvm.git synced 2025-11-18 23:46:31 +01:00
Code Issues Packages Projects Releases Wiki Activity

unify venv calling, add basic huggingchat module

Browse Source
This commit is contained in:
Believethehype
2023-12-29 00:27:34 +01:00
parent 2c8e81d34a
commit 6de23645e4
23 changed files with 168 additions and 200 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
nostr_dvm/tasks/texttospeech.py
View File

@@ -5,7 +5,7 @@ os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
from pathlib import Path
import urllib.request
from nostr_dvm.interfaces.dvmtaskinterface import DVMTaskInterface
from nostr_dvm.interfaces.dvmtaskinterface import DVMTaskInterface, process_venv
from nostr_dvm.utils.admin_utils import AdminConfig
from nostr_dvm.utils.definitions import EventDefinitions
from nostr_dvm.utils.dvmconfig import DVMConfig, build_default_config
@@ -106,8 +106,8 @@ class TextToSpeech(DVMTaskInterface):
print(TTS().list_models())
try:
#model = "tts_models/deu/fairseq/vits"
#model = "tts_models/multilingual/multi-dataset/your_tts"
# model = "tts_models/deu/fairseq/vits"
# model = "tts_models/multilingual/multi-dataset/your_tts"
model = "tts_models/multilingual/multi-dataset/xtts_v2"
tts = TTS(model).to(device)
@@ -154,13 +154,5 @@ def build_example(name, identifier, admin_config):
options=options)
def process_venv():
args = DVMTaskInterface.process_args()
dvm_config = build_default_config(args.identifier)
dvm = TextToSpeech(name="", dvm_config=dvm_config, nip89config=NIP89Config(), admin_config=None)
result = dvm.process(json.loads(args.request))
DVMTaskInterface.write_output(result, args.output)
if __name__ == '__main__':
process_venv()
process_venv(TextToSpeech)