- Add ONNX embedding model for CPU devices, Update the similarithy threshold, improve the embedding speed.

2024-05-19 22:30:10 +08:00
parent bf00c26a83
commit 13a3b21d19
11 changed files with 61433 additions and 49 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -16,3 +16,5 @@ uvicorn==0.29.0
 transformers==4.40.2
 chromedriver-autoinstaller==0.6.4
 torch==2.3.0
+onnxruntime==1.14.1
+tokenizers==0.13.2