Fix deprecated splade model (#333)

joein · web-flow · commit ab9ab73278cd · 2024-08-23T21:14:51.000+02:00
* fix: return prithvida model to supported models

* fix: fix deprecation warning stacklevel
diff --git a/fastembed/sparse/sparse_text_embedding.py b/fastembed/sparse/sparse_text_embedding.py
@@ -56,6 +56,7 @@ def __init__(
                 "The right spelling is prithivida/Splade_PP_en_v1. "
                 "Support of this name will be removed soon, please fix the model_name",
                 DeprecationWarning,
+                stacklevel=2,
             )
             model_name = "prithivida/Splade_PP_en_v1"
 
diff --git a/fastembed/sparse/splade_pp.py b/fastembed/sparse/splade_pp.py
@@ -21,6 +21,16 @@
         },
         "model_file": "model.onnx",
     },
+    {
+        "model": "prithvida/Splade_PP_en_v1",
+        "vocab_size": 30522,
+        "description": "Independent Implementation of SPLADE++ Model for English",
+        "size_in_GB": 0.532,
+        "sources": {
+            "hf": "Qdrant/SPLADE_PP_en_v1",
+        },
+        "model_file": "model.onnx",
+    },
 ]
 
 

Original file line number	Diff line number	Diff line change
`@@ -56,6 +56,7 @@ def __init__(`
`56`	`56`	`"The right spelling is prithivida/Splade_PP_en_v1. "`
`57`	`57`	`"Support of this name will be removed soon, please fix the model_name",`
`58`	`58`	`DeprecationWarning,`
	`59`	`+ stacklevel=2,`
`59`	`60`	`)`
`60`	`61`	`model_name = "prithivida/Splade_PP_en_v1"`
`61`	`62`