Sarp Bilgiç commited on
Commit
60e2e0e
·
1 Parent(s): dac8fbe

model name fix

Browse files
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 = "Qdrant/SPLADE_PP_en_v1"):
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,