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

feat: chat2tts stream

Browse Source
This commit is contained in:
verachen
2025-01-09 11:29:34 +08:00
parent ec3b4b143a
commit 37174413fe
12 changed files with 643 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/blackbox/chroma_upsert.py
View File

@ -33,7 +33,7 @@ class ChromaUpsert(Blackbox):
# load embedding model
self.embedding_model_1 = SentenceTransformerEmbeddings(model_name="/Workspace/Models/BAAI/bge-large-zh-v1.5", model_kwargs={"device": "cuda"})
# load chroma db
self.client_1 = chromadb.HttpClient(host='192.168.0.200', port=7000)
self.client_1 = chromadb.HttpClient(host='10.6.44.141', port=7000)
def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs)