Commit Graph

14 Commits

Author SHA1 Message Date
George
0a4ed42b58 fix: unify existing patterns, remove redundant (#168) 2024-03-30 14:40:19 +05:30
Nirant
d817da2e01 Add Splade v1 (#144)
* Add SPLADE v1

* WIP SPLADE Export errors

* add ONNX model to HF hub and use that

* Update sentences in Converting_SPLADE_to_ONNX.ipynb

* Remove unnecessary files and directories

* Rename var in TextEmbedding class to use EMBEDDING_MODEL_TYPE

* Add SPLADE to list of text embeddings

* Add SPLADE model support for text embedding

* Fix deprecation warning in embedding.py

* Add test for batch embedding with sparse embeddings

* Refactor import statement in test_sparse_embeddings.py

* Rename nbs

* Update vocab size in SPLADE model

* Fix canonical vector lookup in test_text_onnx_embeddings.py

* review refactoring

* restore list_supported_models in OnnxTextEmbedding

* Remove unused method _preprocess_onnx_input() in SpladePP class

* Update SPLADE_PP_en_v1 source in splade_pp.py

* Refactor onnx_model.py to change base model behavior

* extend tests to sparse values as well as indicies

* chore: pre-commit hooks

---------

Co-authored-by: generall <andrey@vasnetsov.com>
Co-authored-by: Anush008 <anushshetty90@gmail.com>
2024-03-13 18:04:44 +05:30
NirantK
206a633eb1 Fix for # https://github.com/pytorch/pytorch/issues/100974 2023-09-25 17:12:03 +05:30
NirantK
0ab82da4aa * chore(.gitignore): add qdrant_storage/ directory to the gitignore file 2023-09-18 17:24:31 +05:30
NirantK
5906ef63d6 * chore(.gitignore): add ignore rule for experimental .bin files in docs/experimental directory 2023-09-18 08:46:58 +05:30
NirantK
399d11eb8a * chore(.gitignore): add rule to ignore experimental .parquet files in docs/experimental directory 2023-08-25 12:56:25 +05:30
NirantK
76cc10c761 * chore(.gitignore): update .gitignore to exclude local_cache directory in multiple locations 2023-08-23 11:06:52 +05:30
NirantK
bd06e3d932 * chore(.gitignore): add docs/local_cache/*/* to ignore list 2023-08-22 11:10:11 +05:30
NirantK
6d33b85f37 * chore(.gitignore): add missing newline at the end of the file
* chore(.gitignore): add nbs/fast-*/* to the ignore list
2023-08-17 12:47:36 +05:30
NirantK
c9b0811b57 * chore(.gitignore): add nbs/local_cache/*/* and local_cache/*/* to ignore list 2023-08-10 19:19:28 +05:30
NirantK
1c684b1208 * chore(.gitignore): add *.tar.gz to the list of ignored files 2023-07-31 16:42:45 +05:30
NirantK
7a23c6a25d * chore(.gitignore): add nbs/*.tar.gz to ignore list 2023-07-31 16:10:42 +05:30
NirantK
1640e6f40e * chore(.gitignore): add .DS_Store to the ignore list 2023-07-31 13:52:42 +05:30
Nirant
d567f27ce8 Initial commit 2023-07-05 08:49:59 +05:30