fahmiaziz98 commited on
Commit
4cf20a7
·
1 Parent(s): 93658bf
Files changed (2) hide show
  1. config.yaml +3 -3
  2. core/embedding.py +1 -1
config.yaml CHANGED
@@ -40,8 +40,8 @@ models:
40
  language: ["multilingual"]
41
  repository: "https://huggingface.co/intfloat/multilingual-e5-small"
42
 
43
- splade-pp-v1:
44
- name: "prithivida/Splade_PP_en_v1"
45
  type: "sparse-embeddings"
46
  dimension: 1234 # must add this field
47
  max_tokens: 1234
@@ -52,7 +52,7 @@ models:
52
  Hereinafter efficiency is a short hand for retrieval efficiency unless explicitly qualified otherwise.
53
  Not that inference efficiency is not important, we will address that subsequently.)
54
  language: ["multilingual"]
55
- repository: "https://huggingface.co/prithivida/Splade_PP_en_v1"
56
 
57
  splade-pp-v2:
58
  name: "prithivida/Splade_PP_en_v2"
 
40
  language: ["multilingual"]
41
  repository: "https://huggingface.co/intfloat/multilingual-e5-small"
42
 
43
+ splade-v3:
44
+ name: "naver/splade-v3"
45
  type: "sparse-embeddings"
46
  dimension: 1234 # must add this field
47
  max_tokens: 1234
 
52
  Hereinafter efficiency is a short hand for retrieval efficiency unless explicitly qualified otherwise.
53
  Not that inference efficiency is not important, we will address that subsequently.)
54
  language: ["multilingual"]
55
+ repository: "https://huggingface.co/naver/splade-v3"
56
 
57
  splade-pp-v2:
58
  name: "prithivida/Splade_PP_en_v2"
core/embedding.py CHANGED
@@ -81,7 +81,7 @@ class SparseEmbeddingModel:
81
 
82
  def _format_values(self, values: List[float]) -> List[float]:
83
  """Format float values to a fixed precision."""
84
- return [round(float(v), 7) for v in values]
85
 
86
  def load(self) -> None:
87
  """Load the sparse embedding model."""
 
81
 
82
  def _format_values(self, values: List[float]) -> List[float]:
83
  """Format float values to a fixed precision."""
84
+ return [round(float(v), 8) for v in values]
85
 
86
  def load(self) -> None:
87
  """Load the sparse embedding model."""