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

vlms updated

Browse Source
This commit is contained in:
gdw6463
2024-05-08 12:50:46 +08:00
parent f3cd6cb09a
commit 2fbceddae5
3 changed files with 77 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/blackbox/blackbox_factory.py
View File

@ -12,6 +12,7 @@ from .text_and_image import TextAndImage
from .chroma_query import ChromaQuery
from .chroma_upsert import ChromaUpsert
from .chroma_chat import ChromaChat
from .vlms import VLMS
from injector import inject, singleton
@singleton
@ -29,10 +30,11 @@ class BlackboxFactory:
fastchat: Fastchat,
audio_chat: AudioChat,
g2e: G2E,
text_and_image:TextAndImage,
text_and_image: TextAndImage,
chroma_query: ChromaQuery,
chroma_upsert: ChromaUpsert,
chroma_chat: ChromaChat) -> None:
chroma_chat: ChromaChat,
vlms: VLMS) -> None:
self.models["audio_to_text"] = audio_to_text
self.models["text_to_audio"] = text_to_audio
self.models["asr"] = asr
@ -46,6 +48,7 @@ class BlackboxFactory:
self.models["chroma_query"] = chroma_query
self.models["chroma_upsert"] = chroma_upsert
self.models["chroma_chat"] = chroma_chat
self.models["vlms"] = vlms
def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs)