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

feat: dotchain

Browse Source
This commit is contained in:
chenyunda218
2024-08-01 11:33:09 +08:00
parent abb5195e55
commit bbf3fe2909
16 changed files with 1571 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/blackbox/blackbox_factory.py
View File

@ -102,6 +102,16 @@ def cosyvoicetts_loader():
from .cosyvoicetts import CosyVoiceTTS
return Injector().get(CosyVoiceTTS)
@model_loader(lazy=blackboxConf.lazyloading)
def workflow_loader():
from .workflow import Workflow
return Injector().get(Workflow)
@model_loader(lazy=blackboxConf.lazyloading)
def sum_loader():
from .sum import Sum
return Injector().get(Sum)
@singleton
class BlackboxFactory:
models = {}
@ -124,6 +134,8 @@ class BlackboxFactory:
self.models["chat"] = chat_loader
self.models["chat_llama"] = chat_llama_loader
self.models["cosyvoicetts"] = cosyvoicetts_loader
self.models["workflow"] = workflow_loader
self.models["sum"] = sum_loader
def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs)