External/agentic-soc-platform
1
0
Fork 0
mirror of https://github.com/FunnyWolf/agentic-soc-platform.git synced 2026-08-22 13:12:56 +02:00
Code Issues Packages Projects Releases Wiki Activity

update Huggingface

Browse Source
This commit is contained in:
rootkit
2025-12-29 03:19:27 +08:00
parent f6ffa17140
commit 9209492f89
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
PLUGINS/Huggingface/__init__.py
View File
Docker/Huggingface/download_model.py → PLUGINS/Huggingface/download_model.py
+2 -2
View File
@@ -15,10 +15,10 @@ from Lib.configs import BASE_DIR
if __name__ == "__main__":
# download reranker model (BAAI/bge-reranker-v2-m3)
model_id = "BAAI/bge-reranker-v2-m3"
local_dir = os.path.join(BASE_DIR, 'Docker', 'Huggingface', 'bge-reranker-v2-m3')
local_dir = os.path.join(BASE_DIR, 'Docker', 'Huggingface', '../../Docker/Huggingface/bge-reranker-v2-m3')
snapshot_download(repo_id=model_id, local_dir=local_dir)
# download sparse model (bm25)
cache_dir = os.path.join(BASE_DIR, 'Docker', 'Huggingface', 'bm25')
cache_dir = os.path.join(BASE_DIR, 'Docker', 'Huggingface', '../../Docker/Huggingface/bm25')
model = SparseTextEmbedding(model_name="Qdrant/bm25", cache_dir=cache_dir)