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

wip: temp

Browse Source
This commit is contained in:
superobk
2024-05-16 16:02:13 +08:00
parent e4136ee4e3
commit 0612c065c0
4 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/blackbox/blackbox_factory.py
View File

@ -37,7 +37,6 @@ class BlackboxFactory:
#chroma_query: ChromaQuery,
#chroma_upsert: ChromaUpsert,
#chroma_chat: ChromaChat,
melotts: MeloTTS,
vlms: VLMS) -> None:
self.models["audio_to_text"] = audio_to_text
self.models["text_to_audio"] = text_to_audio
@ -50,10 +49,9 @@ class BlackboxFactory:
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["melotts"] = melotts
#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):

1
src/blackbox/g2e.py
View File

@ -65,6 +65,7 @@ class G2E(Blackbox):
api_key='YOUR_API_KEY',
base_url=url
)
#model_name = client.models.list().data[0].id
model_name = client.models.list().data[1].id
print(model_name)
response = client.chat.completions.create(