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

feat: cosyvoicetts

Browse Source
This commit is contained in:
chenyunda218
2024-07-29 10:45:38 +08:00
parent 0315955a38
commit 05826d2e93
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/blackbox/blackbox_factory.py
View File

@ -97,6 +97,11 @@ def chat_llama_loader():
from .chat_llama import ChatLLaMA from .chat_llama import ChatLLaMA
return Injector().get(ChatLLaMA) return Injector().get(ChatLLaMA)
@model_loader(lazy=blackboxConf.lazyloading)
def cosyvoicetts_loader():
from .cosyvoicetts import CosyVoiceTTS
return Injector().get(CosyVoiceTTS)
@singleton @singleton
class BlackboxFactory: class BlackboxFactory:
models = {} models = {}
@ -118,6 +123,7 @@ class BlackboxFactory:
self.models["vlms"] = vlms_loader self.models["vlms"] = vlms_loader
self.models["chat"] = chat_loader self.models["chat"] = chat_loader
self.models["chat_llama"] = chat_llama_loader self.models["chat_llama"] = chat_llama_loader
self.models["cosyvoicetts"] = cosyvoicetts_loader
def __call__(self, *args, **kwargs): def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs) return self.processing(*args, **kwargs)