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

Merge branch 'vera' into main

Browse Source
This commit is contained in:
0Xiao0
2024-08-22 16:00:12 +08:00
parent 90f1601e66 c247d611a0
commit cebf0aca74
28 changed files with 692 additions and 205 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
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():
@ -141,11 +141,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
self.models["workflow"] = workflow_loader
self.models["sum"] = sum_loader