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: blackbox chat_llama updated

Browse Source
This commit is contained in:
0Xiao0
2024-06-02 01:11:21 +08:00
parent b3f8de2393
commit a96e845807
4 changed files with 6567 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/blackbox/blackbox_factory.py
View File

@ -67,11 +67,6 @@ def fastchat_loader():
from .fastchat import Fastchat
return Injector().get(Fastchat)
@model_loader(lazy=blackboxConf.lazyloading)
def chat_loader():
from .chat import Chat
return Injector().get(Chat)
@model_loader(lazy=blackboxConf.lazyloading)
def chroma_query_loader():
from .chroma_query import ChromaQuery
@ -83,10 +78,20 @@ def chroma_upsert_loader():
return Injector().get(ChromaUpsert)
@model_loader(lazy=blackboxConf.lazyloading)
def chroma_chat_load():
def chroma_chat_loader():
from .chroma_chat import ChromaChat
return Injector().get(ChromaChat)
@model_loader(lazy=blackboxConf.lazyloading)
def chat_loader():
from .chat import Chat
return Injector().get(Chat)
@model_loader(lazy=blackboxConf.lazyloading)
def chat_llama_loader():
from .chat_llama import ChatLLaMA
return Injector().get(ChatLLaMA)
@singleton
class BlackboxFactory:
models = {}
@ -103,10 +108,11 @@ class BlackboxFactory:
self.models["text_and_image"] = text_and_image_loader
self.models["chroma_query"] = chroma_query_loader
self.models["chroma_upsert"] = chroma_upsert_loader
self.models["chroma_chat"] = chroma_chat_load
self.models["chroma_chat"] = chroma_chat_loader
self.models["melotts"] = melotts_loader
self.models["vlms"] = vlms_loader
self.models["chat"] = chat_loader
self.models["chat_llama"] = chat_llama_loader
def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs)