Spaces:
Build error
Build error
Sarp Bilgiç commited on
Commit ·
60e2e0e
1
Parent(s): dac8fbe
model name fix
Browse files- requirements-dev.txt +1 -1
- requirements-prod.txt +1 -1
- src/clients/sparse_embedding_client.py +1 -1
requirements-dev.txt
CHANGED
|
@@ -30,4 +30,4 @@ tenacity
|
|
| 30 |
#llama-index-embeddings-huggingface
|
| 31 |
#transformers
|
| 32 |
llama-index-embeddings-fastembed
|
| 33 |
-
fastembed
|
|
|
|
| 30 |
#llama-index-embeddings-huggingface
|
| 31 |
#transformers
|
| 32 |
llama-index-embeddings-fastembed
|
| 33 |
+
fastembed
|
requirements-prod.txt
CHANGED
|
@@ -24,4 +24,4 @@ llama-index-tools-tavily-research
|
|
| 24 |
qdrant-client
|
| 25 |
tenacity
|
| 26 |
llama-index-embeddings-fastembed
|
| 27 |
-
fastembed
|
|
|
|
| 24 |
qdrant-client
|
| 25 |
tenacity
|
| 26 |
llama-index-embeddings-fastembed
|
| 27 |
+
fastembed
|
src/clients/sparse_embedding_client.py
CHANGED
|
@@ -3,7 +3,7 @@ from typing import List, Tuple
|
|
| 3 |
import logging
|
| 4 |
|
| 5 |
class SparseEmbeddingClient:
|
| 6 |
-
def __init__(self, model_name: str = "
|
| 7 |
logging.info(f"Initializing SPLADE sparse embeddings ({model_name})...")
|
| 8 |
self.model = SparseTextEmbedding(
|
| 9 |
model_name=model_name,
|
|
|
|
| 3 |
import logging
|
| 4 |
|
| 5 |
class SparseEmbeddingClient:
|
| 6 |
+
def __init__(self, model_name: str = "prithivida/Splade_PP_en_v1"):
|
| 7 |
logging.info(f"Initializing SPLADE sparse embeddings ({model_name})...")
|
| 8 |
self.model = SparseTextEmbedding(
|
| 9 |
model_name=model_name,
|