4 Commits

Author	SHA1	Message	Date
George	e11f66bb55	refactoring: update imports in notebooks (#173) ... * new: simplify imports * refactoring: update import * refactoring: update imports in notebooks * fix: fix notebook output * Re-run notebook with revised imports --------- Co-authored-by: Nirant Kasliwal <nirant.bits@gmail.com>	2024-04-01 17:02:13 +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
Nirant	74062e8607	Add attention export functionality to experiments (#134) ... * Add attention export functionality * Update experiments/attention_export.py Co-authored-by: Anush <anushshetty90@gmail.com> --------- Co-authored-by: Anush <anushshetty90@gmail.com>	2024-03-04 16:30:17 +05:30
Kumar Shivendu	05885a36dd	refactor: Introduce experiments dir (#108)	2024-02-05 10:49:12 +05:30