ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-14 17:13:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

add asr and tts with settings

Browse Source
This commit is contained in:
0Xiao0
2024-08-22 15:26:27 +08:00
parent abb5195e55
commit c247d611a0
28 changed files with 626 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/blackbox/blackbox_factory.py
View File

@ -47,15 +47,15 @@ def vlms_loader():
from .vlms import VLMS
return Injector().get(VLMS)
@model_loader(lazy=blackboxConf.lazyloading)
def melotts_loader():
from .melotts import MeloTTS
return Injector().get(MeloTTS)
# @model_loader(lazy=blackboxConf.lazyloading)
# def melotts_loader():
# from .melotts import MeloTTS
# return Injector().get(MeloTTS)
@model_loader(lazy=blackboxConf.lazyloading)
def cosyvoicetts_loader():
from .cosyvoicetts import CosyVoiceTTS
return Injector().get(CosyVoiceTTS)
# @model_loader(lazy=blackboxConf.lazyloading)
# def cosyvoicetts_loader():
# from .cosyvoicetts import CosyVoiceTTS
# return Injector().get(CosyVoiceTTS)
@model_loader(lazy=blackboxConf.lazyloading)
def tts_loader():
@ -97,11 +97,6 @@ def chat_llama_loader():
from .chat_llama import ChatLLaMA
return Injector().get(ChatLLaMA)
@model_loader(lazy=blackboxConf.lazyloading)
def cosyvoicetts_loader():
from .cosyvoicetts import CosyVoiceTTS
return Injector().get(CosyVoiceTTS)
@singleton
class BlackboxFactory:
models = {}
@ -119,11 +114,11 @@ class BlackboxFactory:
self.models["chroma_query"] = chroma_query_loader
self.models["chroma_upsert"] = chroma_upsert_loader
self.models["chroma_chat"] = chroma_chat_loader
self.models["melotts"] = melotts_loader
# self.models["melotts"] = melotts_loader
self.models["vlms"] = vlms_loader
self.models["chat"] = chat_loader
self.models["chat_llama"] = chat_llama_loader
self.models["cosyvoicetts"] = cosyvoicetts_loader
# self.models["cosyvoicetts"] = cosyvoicetts_loader
def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs)