upstream-archive/fastembed
1
0
Fork 0
mirror of https://github.com/qdrant/fastembed.git synced 2026-08-03 08:30:58 -05:00
Code Issues Packages Projects Releases Wiki Activity

* feat(embedding.py): add support for custom model name, onnx providers, and max length in DefaultEmbedding constructor

Browse Source
This commit is contained in:
NirantK
2023-08-10 10:43:50 +05:30
parent 6cce4e0b9a
commit 885a738751
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
fastembed/embedding.py
View File

@@ -37,7 +37,10 @@ class ONNXProviders:
Metal = "CoreMLExecutionProvider"
class DefaultEmbedding(Embedding):
def __init__(self, model_name: str = "sentence-transformers/all-MiniLM-L6-v2"):
def __init__(self,
model_name: str = "BAAI/bge-base-en",
onnx_providers: List[str] = [ONNXProviders.Metal],
max_length: int = 512):
self.cache_dir = Path(tempfile.gettempdir()) / "fastembed"
filepath = self.download_file_from_gcs(
"https://storage.googleapis.com/qdrant-fastembed/fast-all-MiniLM-L6-v2.tar.gz",