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

tmp

Browse Source
This commit is contained in:
superobk
2024-05-16 17:05:11 +08:00
parent a586d97c12
commit 032f6cb35a
3 changed files with 15 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/blackbox/blackbox_factory.py
View File

@ -3,16 +3,16 @@ from .sentiment import Sentiment
from .tts import TTS
from .asr import ASR
from .audio_to_text import AudioToText
from .emotion import Emotion
#from .emotion import Emotion
from .blackbox import Blackbox
from .text_to_audio import TextToAudio
from .tesou import Tesou
from .fastchat import Fastchat
from .g2e import G2E
from .text_and_image import TextAndImage
from .chroma_query import ChromaQuery
from .chroma_upsert import ChromaUpsert
from .chroma_chat import ChromaChat
# from .chroma_query import ChromaQuery
# from .chroma_upsert import ChromaUpsert
# from .chroma_chat import ChromaChat
from .melotts import MeloTTS
from .vlms import VLMS
from injector import inject, singleton
@ -28,15 +28,15 @@ class BlackboxFactory:
asr: ASR,
tts: TTS,
sentiment_engine: Sentiment,
emotion: Emotion,
#emotion: Emotion,
tesou: Tesou,
fastchat: Fastchat,
audio_chat: AudioChat,
g2e: G2E,
text_and_image: TextAndImage,
chroma_query: ChromaQuery,
chroma_upsert: ChromaUpsert,
chroma_chat: ChromaChat,
#chroma_query: ChromaQuery,
#chroma_upsert: ChromaUpsert,
#chroma_chat: ChromaChat,
melotts: MeloTTS,
vlms: VLMS) -> None:
self.models["audio_to_text"] = audio_to_text
@ -45,14 +45,14 @@ class BlackboxFactory:
self.models["tts"] = tts
self.models["sentiment_engine"] = sentiment_engine
self.models["tesou"] = tesou
self.models["emotion"] = emotion
#self.models["emotion"] = emotion
self.models["fastchat"] = fastchat
self.models["audio_chat"] = audio_chat
self.models["g2e"] = g2e
self.models["text_and_image"] = text_and_image
self.models["chroma_query"] = chroma_query
self.models["chroma_upsert"] = chroma_upsert
self.models["chroma_chat"] = chroma_chat
# self.models["chroma_query"] = chroma_query
# self.models["chroma_upsert"] = chroma_upsert
# self.models["chroma_chat"] = chroma_chat
self.models["melotts"] = melotts
self.models["vlms"] = vlms