Skip to content

Commit d79d165

Browse files
authored
Add new example HF embedding models (#415)
from https://huggingface.co/models?library=sentence-transformers
1 parent 8ebd8d7 commit d79d165

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

helpers/providers/huggingface.ts

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,13 @@ type ModelData = {
88
};
99
const EMBEDDING_MODELS: Record<string, ModelData> = {
1010
"BAAI/bge-small-en-v1.5": { dimensions: 384 },
11+
"BAAI/bge-base-en-v1.5": { dimensions: 768 },
12+
"BAAI/bge-large-en-v1.5": { dimensions: 1024 },
13+
"sentence-transformers/all-MiniLM-L6-v2": { dimensions: 384 },
14+
"sentence-transformers/all-mpnet-base-v2": { dimensions: 768 },
15+
"intfloat/multilingual-e5-large": { dimensions: 1024 },
16+
"mixedbread-ai/mxbai-embed-large-v1": { dimensions: 1024 },
17+
"nomic-ai/nomic-embed-text-v1.5": { dimensions: 768 },
1118
};
1219

1320
const DEFAULT_MODEL = MODELS[0];

0 commit comments

Comments
 (0)