diff --git a/helpers/providers/huggingface.ts b/helpers/providers/huggingface.ts index 1a3a4a069bfe1945d86bbd3410a7e681f3061574..039b8e323048f5d50320a46f26a32125b1354138 100644 --- a/helpers/providers/huggingface.ts +++ b/helpers/providers/huggingface.ts @@ -8,6 +8,13 @@ type ModelData = { }; const EMBEDDING_MODELS: Record<string, ModelData> = { "BAAI/bge-small-en-v1.5": { dimensions: 384 }, + "BAAI/bge-base-en-v1.5": { dimensions: 768 }, + "BAAI/bge-large-en-v1.5": { dimensions: 1024 }, + "sentence-transformers/all-MiniLM-L6-v2": { dimensions: 384 }, + "sentence-transformers/all-mpnet-base-v2": { dimensions: 768 }, + "intfloat/multilingual-e5-large": { dimensions: 1024 }, + "mixedbread-ai/mxbai-embed-large-v1": { dimensions: 1024 }, + "nomic-ai/nomic-embed-text-v1.5": { dimensions: 768 }, }; const DEFAULT_MODEL = MODELS[0];