gitextract_by8kvk5i/ ├── .github/ │ └── workflows/ │ ├── quality.yml │ └── tests.yml ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── MANIFEST.in ├── Makefile ├── NOTICE.txt ├── README.md ├── docs/ │ ├── .htaccess │ ├── Makefile │ ├── _static/ │ │ ├── css/ │ │ │ └── custom.css │ │ ├── html/ │ │ │ └── models_en_sentence_embeddings.html │ │ └── js/ │ │ └── custom.js │ ├── _templates/ │ │ └── layout.html │ ├── conf.py │ ├── cross_encoder/ │ │ ├── loss_overview.md │ │ ├── pretrained_models.md │ │ ├── training/ │ │ │ └── examples.rst │ │ ├── training_overview.md │ │ └── usage/ │ │ ├── efficiency.rst │ │ └── usage.rst │ ├── img/ │ │ └── logo.xcf │ ├── installation.md │ ├── migration_guide.md │ ├── package_reference/ │ │ ├── cross_encoder/ │ │ │ ├── cross_encoder.md │ │ │ ├── evaluation.md │ │ │ ├── index.rst │ │ │ ├── losses.md │ │ │ ├── trainer.md │ │ │ └── training_args.md │ │ ├── sentence_transformer/ │ │ │ ├── SentenceTransformer.md │ │ │ ├── datasets.md │ │ │ ├── evaluation.md │ │ │ ├── index.rst │ │ │ ├── losses.md │ │ │ ├── models.md │ │ │ ├── quantization.md │ │ │ ├── sampler.md │ │ │ ├── trainer.md │ │ │ └── training_args.md │ │ ├── sparse_encoder/ │ │ │ ├── SparseEncoder.md │ │ │ ├── callbacks.md │ │ │ ├── evaluation.md │ │ │ ├── index.rst │ │ │ ├── losses.md │ │ │ ├── models.md │ │ │ ├── search_engines.md │ │ │ ├── trainer.md │ │ │ └── training_args.md │ │ └── util.md │ ├── pretrained-models/ │ │ ├── ce-msmarco.md │ │ ├── dpr.md │ │ ├── msmarco-v1.md │ │ ├── msmarco-v2.md │ │ ├── msmarco-v3.md │ │ ├── msmarco-v5.md │ │ ├── nli-models.md │ │ ├── nq-v1.md │ │ ├── sts-models.md │ │ └── wikipedia-sections-models.md │ ├── publications.md │ ├── quickstart.rst │ ├── requirements.txt │ ├── sentence_transformer/ │ │ ├── dataset_overview.md │ │ ├── loss_overview.md │ │ ├── pretrained_models.md │ │ ├── training/ │ │ │ ├── distributed.rst │ │ │ └── examples.rst │ │ ├── training_overview.md │ │ └── usage/ │ │ ├── backend_export_sidebar.rst │ │ ├── custom_models.rst │ │ ├── efficiency.rst │ │ ├── mteb_evaluation.md │ │ ├── semantic_textual_similarity.rst │ │ └── usage.rst │ └── sparse_encoder/ │ ├── loss_overview.md │ ├── pretrained_models.md │ ├── training/ │ │ └── examples.rst │ ├── training_overview.md │ └── usage/ │ ├── efficiency.rst │ └── usage.rst ├── examples/ │ ├── cross_encoder/ │ │ ├── applications/ │ │ │ ├── README.md │ │ │ ├── cross-encoder_reranking.py │ │ │ └── cross-encoder_usage.py │ │ └── training/ │ │ ├── README.md │ │ ├── distillation/ │ │ │ ├── README.md │ │ │ ├── train_cross_encoder_kd_margin_mse.py │ │ │ └── train_cross_encoder_kd_mse.py │ │ ├── ms_marco/ │ │ │ ├── README.md │ │ │ ├── eval_cross-encoder-trec-dl.py │ │ │ ├── training_ms_marco_bce.py │ │ │ ├── training_ms_marco_bce_preprocessed.py │ │ │ ├── training_ms_marco_cmnrl.py │ │ │ ├── training_ms_marco_lambda.py │ │ │ ├── training_ms_marco_lambda_hard_neg.py │ │ │ ├── training_ms_marco_lambda_preprocessed.py │ │ │ ├── training_ms_marco_listmle.py │ │ │ ├── training_ms_marco_listnet.py │ │ │ ├── training_ms_marco_plistmle.py │ │ │ └── training_ms_marco_ranknet.py │ │ ├── nli/ │ │ │ ├── README.md │ │ │ └── training_nli.py │ │ ├── quora_duplicate_questions/ │ │ │ ├── README.md │ │ │ └── training_quora_duplicate_questions.py │ │ ├── rerankers/ │ │ │ ├── README.md │ │ │ ├── training_gooaq_bce.py │ │ │ ├── training_gooaq_cmnrl.py │ │ │ ├── training_gooaq_lambda.py │ │ │ └── training_nq_bce.py │ │ └── sts/ │ │ ├── README.md │ │ └── training_stsbenchmark.py │ ├── sentence_transformer/ │ │ ├── README.md │ │ ├── applications/ │ │ │ ├── README.md │ │ │ ├── clustering/ │ │ │ │ ├── README.md │ │ │ │ ├── agglomerative.py │ │ │ │ ├── fast_clustering.py │ │ │ │ └── kmeans.py │ │ │ ├── computing-embeddings/ │ │ │ │ ├── README.rst │ │ │ │ ├── computing_embeddings.py │ │ │ │ ├── computing_embeddings_multi_gpu.py │ │ │ │ └── computing_embeddings_streaming.py │ │ │ ├── embedding-quantization/ │ │ │ │ ├── README.md │ │ │ │ ├── semantic_search_faiss.py │ │ │ │ ├── semantic_search_faiss_benchmark.py │ │ │ │ ├── semantic_search_recommended.py │ │ │ │ ├── semantic_search_usearch.py │ │ │ │ └── semantic_search_usearch_benchmark.py │ │ │ ├── image-search/ │ │ │ │ ├── Image_Classification.ipynb │ │ │ │ ├── Image_Clustering.ipynb │ │ │ │ ├── Image_Duplicates.ipynb │ │ │ │ ├── Image_Search-multilingual.ipynb │ │ │ │ ├── Image_Search.ipynb │ │ │ │ ├── README.md │ │ │ │ └── example.py │ │ │ ├── parallel-sentence-mining/ │ │ │ │ ├── README.md │ │ │ │ ├── bitext_mining.py │ │ │ │ ├── bitext_mining_utils.py │ │ │ │ └── bucc2018.py │ │ │ ├── paraphrase-mining/ │ │ │ │ └── README.md │ │ │ ├── retrieve_rerank/ │ │ │ │ ├── README.md │ │ │ │ ├── in_document_search_crossencoder.py │ │ │ │ └── retrieve_rerank_simple_wikipedia.ipynb │ │ │ ├── semantic-search/ │ │ │ │ ├── README.md │ │ │ │ ├── semantic_search.py │ │ │ │ ├── semantic_search_nq_opensearch.py │ │ │ │ ├── semantic_search_publications.py │ │ │ │ ├── semantic_search_quora_annoy.py │ │ │ │ ├── semantic_search_quora_elasticsearch.py │ │ │ │ ├── semantic_search_quora_faiss.py │ │ │ │ ├── semantic_search_quora_hnswlib.py │ │ │ │ ├── semantic_search_quora_pytorch.py │ │ │ │ └── semantic_search_wikipedia_qa.py │ │ │ └── text-summarization/ │ │ │ ├── LexRank.py │ │ │ ├── README.md │ │ │ └── text-summarization.py │ │ ├── domain_adaptation/ │ │ │ └── README.md │ │ ├── evaluation/ │ │ │ ├── evaluation_inference_speed.py │ │ │ ├── evaluation_no_dup_batch_sampler_speed.py │ │ │ ├── evaluation_stsbenchmark.py │ │ │ └── evaluation_translation_matching.py │ │ ├── training/ │ │ │ ├── README.md │ │ │ ├── adaptive_layer/ │ │ │ │ ├── README.md │ │ │ │ ├── adaptive_layer_nli.py │ │ │ │ └── adaptive_layer_sts.py │ │ │ ├── avg_word_embeddings/ │ │ │ │ ├── training_stsbenchmark_avg_word_embeddings.py │ │ │ │ ├── training_stsbenchmark_bilstm.py │ │ │ │ ├── training_stsbenchmark_bow.py │ │ │ │ ├── training_stsbenchmark_cnn.py │ │ │ │ └── training_stsbenchmark_tf-idf_word_embeddings.py │ │ │ ├── clip/ │ │ │ │ ├── train_clip.ipynb │ │ │ │ └── training_clip_flickr8k_mlflow.py │ │ │ ├── data_augmentation/ │ │ │ │ ├── README.md │ │ │ │ ├── train_sts_indomain_bm25.py │ │ │ │ ├── train_sts_indomain_nlpaug.py │ │ │ │ ├── train_sts_indomain_semantic.py │ │ │ │ ├── train_sts_qqp_crossdomain.py │ │ │ │ └── train_sts_seed_optimization.py │ │ │ ├── distillation/ │ │ │ │ ├── README.md │ │ │ │ ├── dimensionality_reduction.py │ │ │ │ ├── model_distillation.py │ │ │ │ ├── model_distillation_layer_reduction.py │ │ │ │ └── model_quantization.py │ │ │ ├── hpo/ │ │ │ │ ├── README.rst │ │ │ │ └── hpo_nli.py │ │ │ ├── matryoshka/ │ │ │ │ ├── 2d_matryoshka_nli.py │ │ │ │ ├── 2d_matryoshka_sts.py │ │ │ │ ├── README.md │ │ │ │ ├── matryoshka_eval_stsb.py │ │ │ │ ├── matryoshka_nli.py │ │ │ │ ├── matryoshka_nli_reduced_dim.py │ │ │ │ └── matryoshka_sts.py │ │ │ ├── ms_marco/ │ │ │ │ ├── README.md │ │ │ │ ├── eval_msmarco.py │ │ │ │ ├── multilingual/ │ │ │ │ │ ├── README.md │ │ │ │ │ └── translate_queries.py │ │ │ │ ├── train-kldiv.py │ │ │ │ ├── train-margin-mse.py │ │ │ │ ├── train_bi-encoder_margin-mse.py │ │ │ │ └── train_bi-encoder_mnrl.py │ │ │ ├── multilingual/ │ │ │ │ ├── README.md │ │ │ │ ├── get_parallel_data_opus.py │ │ │ │ ├── get_parallel_data_talks.py │ │ │ │ ├── get_parallel_data_tatoeba.py │ │ │ │ ├── get_parallel_data_wikimatrix.py │ │ │ │ └── make_multilingual.py │ │ │ ├── nli/ │ │ │ │ ├── README.md │ │ │ │ ├── training_nli.py │ │ │ │ ├── training_nli_angle.py │ │ │ │ ├── training_nli_v2.py │ │ │ │ └── training_nli_v3.py │ │ │ ├── other/ │ │ │ │ ├── training_batch_hard_trec.py │ │ │ │ ├── training_gooaq_infonce_gor.py │ │ │ │ ├── training_multi-task.py │ │ │ │ └── training_wikipedia_sections.py │ │ │ ├── paraphrases/ │ │ │ │ ├── README.md │ │ │ │ └── training.py │ │ │ ├── peft/ │ │ │ │ ├── README.md │ │ │ │ └── training_gooaq_lora.py │ │ │ ├── prompts/ │ │ │ │ ├── README.md │ │ │ │ └── training_nq_prompts.py │ │ │ ├── quora_duplicate_questions/ │ │ │ │ ├── README.md │ │ │ │ ├── application_duplicate_questions_mining.py │ │ │ │ ├── create_splits.py │ │ │ │ ├── training_MultipleNegativesRankingLoss.py │ │ │ │ ├── training_OnlineContrastiveLoss.py │ │ │ │ └── training_multi-task-learning.py │ │ │ ├── sts/ │ │ │ │ ├── README.md │ │ │ │ ├── training_stsbenchmark.py │ │ │ │ └── training_stsbenchmark_continue_training.py │ │ │ └── unsloth/ │ │ │ ├── README.md │ │ │ ├── training_gooaq_unsloth.py │ │ │ └── training_medical_unsloth.py │ │ └── unsupervised_learning/ │ │ ├── CT/ │ │ │ ├── README.md │ │ │ ├── train_askubuntu_ct.py │ │ │ ├── train_ct_from_file.py │ │ │ └── train_stsb_ct.py │ │ ├── CT_In-Batch_Negatives/ │ │ │ ├── README.md │ │ │ ├── train_askubuntu_ct-improved.py │ │ │ ├── train_ct-improved_from_file.py │ │ │ └── train_stsb_ct-improved.py │ │ ├── MLM/ │ │ │ ├── README.md │ │ │ └── train_mlm.py │ │ ├── README.md │ │ ├── SimCSE/ │ │ │ ├── README.md │ │ │ ├── train_askubuntu_simcse.py │ │ │ ├── train_simcse_from_file.py │ │ │ └── train_stsb_simcse.py │ │ ├── TSDAE/ │ │ │ ├── README.md │ │ │ ├── eval_askubuntu.py │ │ │ ├── train_askubuntu_tsdae.py │ │ │ ├── train_stsb_tsdae.py │ │ │ └── train_tsdae_from_file.py │ │ └── query_generation/ │ │ ├── 1_programming_query_generation.py │ │ ├── 2_programming_train_bi-encoder.py │ │ ├── 3_programming_semantic_search.py │ │ ├── README.md │ │ └── example_query_generation.py │ └── sparse_encoder/ │ ├── applications/ │ │ ├── README.md │ │ ├── computing_embeddings/ │ │ │ ├── README.rst │ │ │ └── compute_embeddings.py │ │ ├── retrieve_rerank/ │ │ │ ├── README.md │ │ │ ├── hybrid_search.py │ │ │ └── retrieve_rerank_simple_wikipedia.ipynb │ │ ├── semantic_search/ │ │ │ ├── README.md │ │ │ ├── semantic_search_elasticsearch.py │ │ │ ├── semantic_search_manual.py │ │ │ ├── semantic_search_opensearch.py │ │ │ ├── semantic_search_qdrant.py │ │ │ ├── semantic_search_seismic.py │ │ │ └── semantic_search_splade_index.py │ │ └── semantic_textual_similarity/ │ │ ├── README.md │ │ └── semantic_textual_similarity.py │ ├── evaluation/ │ │ ├── README.md │ │ ├── sparse_classification_evaluator.py │ │ ├── sparse_mse_evaluator.py │ │ ├── sparse_nanobeir_advanced_evaluator.py │ │ ├── sparse_nanobeir_evaluator.py │ │ ├── sparse_reranking_evaluator.py │ │ ├── sparse_retrieval_evaluator.py │ │ ├── sparse_similarity_evaluator.py │ │ ├── sparse_translation_evaluator.py │ │ └── sparse_triplet_evaluator.py │ └── training/ │ ├── README.md │ ├── distillation/ │ │ ├── README.md │ │ └── train_splade_msmarco_margin_mse.py │ ├── ms_marco/ │ │ ├── README.md │ │ └── train_splade_msmarco_mnrl.py │ ├── nli/ │ │ ├── README.md │ │ └── train_splade_nli.py │ ├── peft/ │ │ └── train_splade_gooaq_peft.py │ ├── quora_duplicate_questions/ │ │ ├── README.md │ │ └── training_splade_quora.py │ ├── retrievers/ │ │ ├── README.md │ │ ├── train_csr_nq.py │ │ ├── train_splade_gooaq.py │ │ ├── train_splade_nq.py │ │ └── train_splade_nq_cached.py │ └── sts/ │ ├── README.md │ └── train_splade_stsbenchmark.py ├── index.rst ├── pyproject.toml ├── sentence_transformers/ │ ├── LoggingHandler.py │ ├── SentenceTransformer.py │ ├── __init__.py │ ├── backend/ │ │ ├── __init__.py │ │ ├── load.py │ │ ├── optimize.py │ │ ├── quantize.py │ │ └── utils.py │ ├── cross_encoder/ │ │ ├── CrossEncoder.py │ │ ├── __init__.py │ │ ├── data_collator.py │ │ ├── evaluation/ │ │ │ ├── __init__.py │ │ │ ├── classification.py │ │ │ ├── correlation.py │ │ │ ├── deprecated.py │ │ │ ├── nano_beir.py │ │ │ └── reranking.py │ │ ├── fit_mixin.py │ │ ├── losses/ │ │ │ ├── BinaryCrossEntropyLoss.py │ │ │ ├── CachedMultipleNegativesRankingLoss.py │ │ │ ├── CrossEntropyLoss.py │ │ │ ├── LambdaLoss.py │ │ │ ├── ListMLELoss.py │ │ │ ├── ListNetLoss.py │ │ │ ├── MSELoss.py │ │ │ ├── MarginMSELoss.py │ │ │ ├── MultipleNegativesRankingLoss.py │ │ │ ├── PListMLELoss.py │ │ │ ├── RankNetLoss.py │ │ │ └── __init__.py │ │ ├── model_card.py │ │ ├── model_card_template.md │ │ ├── trainer.py │ │ ├── training_args.py │ │ └── util.py │ ├── data_collator.py │ ├── datasets/ │ │ ├── DenoisingAutoEncoderDataset.py │ │ ├── NoDuplicatesDataLoader.py │ │ ├── ParallelSentencesDataset.py │ │ ├── SentenceLabelDataset.py │ │ ├── SentencesDataset.py │ │ └── __init__.py │ ├── evaluation/ │ │ ├── BinaryClassificationEvaluator.py │ │ ├── EmbeddingSimilarityEvaluator.py │ │ ├── InformationRetrievalEvaluator.py │ │ ├── LabelAccuracyEvaluator.py │ │ ├── MSEEvaluator.py │ │ ├── MSEEvaluatorFromDataFrame.py │ │ ├── NanoBEIREvaluator.py │ │ ├── ParaphraseMiningEvaluator.py │ │ ├── RerankingEvaluator.py │ │ ├── SentenceEvaluator.py │ │ ├── SequentialEvaluator.py │ │ ├── SimilarityFunction.py │ │ ├── TranslationEvaluator.py │ │ ├── TripletEvaluator.py │ │ └── __init__.py │ ├── fit_mixin.py │ ├── losses/ │ │ ├── AdaptiveLayerLoss.py │ │ ├── AnglELoss.py │ │ ├── BatchAllTripletLoss.py │ │ ├── BatchHardSoftMarginTripletLoss.py │ │ ├── BatchHardTripletLoss.py │ │ ├── BatchSemiHardTripletLoss.py │ │ ├── CachedGISTEmbedLoss.py │ │ ├── CachedMultipleNegativesRankingLoss.py │ │ ├── CachedMultipleNegativesSymmetricRankingLoss.py │ │ ├── CoSENTLoss.py │ │ ├── ContrastiveLoss.py │ │ ├── ContrastiveTensionLoss.py │ │ ├── CosineSimilarityLoss.py │ │ ├── DenoisingAutoEncoderLoss.py │ │ ├── DistillKLDivLoss.py │ │ ├── GISTEmbedLoss.py │ │ ├── GlobalOrthogonalRegularizationLoss.py │ │ ├── MSELoss.py │ │ ├── MarginMSELoss.py │ │ ├── Matryoshka2dLoss.py │ │ ├── MatryoshkaLoss.py │ │ ├── MegaBatchMarginLoss.py │ │ ├── MultipleNegativesRankingLoss.py │ │ ├── MultipleNegativesSymmetricRankingLoss.py │ │ ├── OnlineContrastiveLoss.py │ │ ├── SoftmaxLoss.py │ │ ├── TripletLoss.py │ │ └── __init__.py │ ├── model_card.py │ ├── model_card_template.md │ ├── model_card_templates.py │ ├── models/ │ │ ├── BoW.py │ │ ├── CLIPModel.py │ │ ├── CNN.py │ │ ├── Dense.py │ │ ├── Dropout.py │ │ ├── InputModule.py │ │ ├── LSTM.py │ │ ├── LayerNorm.py │ │ ├── Module.py │ │ ├── Normalize.py │ │ ├── Pooling.py │ │ ├── Router.py │ │ ├── StaticEmbedding.py │ │ ├── Transformer.py │ │ ├── WeightedLayerPooling.py │ │ ├── WordEmbeddings.py │ │ ├── WordWeights.py │ │ ├── __init__.py │ │ └── tokenizer/ │ │ ├── PhraseTokenizer.py │ │ ├── WhitespaceTokenizer.py │ │ ├── WordTokenizer.py │ │ └── __init__.py │ ├── peft_mixin.py │ ├── py.typed │ ├── quantization.py │ ├── readers/ │ │ ├── InputExample.py │ │ ├── LabelSentenceReader.py │ │ ├── NLIDataReader.py │ │ ├── PairedFilesReader.py │ │ ├── STSDataReader.py │ │ ├── TripletReader.py │ │ └── __init__.py │ ├── sampler.py │ ├── similarity_functions.py │ ├── sparse_encoder/ │ │ ├── SparseEncoder.py │ │ ├── __init__.py │ │ ├── callbacks/ │ │ │ ├── __init__.py │ │ │ └── splade_callbacks.py │ │ ├── data_collator.py │ │ ├── evaluation/ │ │ │ ├── ReciprocalRankFusionEvaluator.py │ │ │ ├── SparseBinaryClassificationEvaluator.py │ │ │ ├── SparseEmbeddingSimilarityEvaluator.py │ │ │ ├── SparseInformationRetrievalEvaluator.py │ │ │ ├── SparseMSEEvaluator.py │ │ │ ├── SparseNanoBEIREvaluator.py │ │ │ ├── SparseRerankingEvaluator.py │ │ │ ├── SparseTranslationEvaluator.py │ │ │ ├── SparseTripletEvaluator.py │ │ │ └── __init__.py │ │ ├── losses/ │ │ │ ├── CSRLoss.py │ │ │ ├── CachedSpladeLoss.py │ │ │ ├── FlopsLoss.py │ │ │ ├── SparseAnglELoss.py │ │ │ ├── SparseCoSENTLoss.py │ │ │ ├── SparseCosineSimilarityLoss.py │ │ │ ├── SparseDistillKLDivLoss.py │ │ │ ├── SparseMSELoss.py │ │ │ ├── SparseMarginMSELoss.py │ │ │ ├── SparseMultipleNegativesRankingLoss.py │ │ │ ├── SparseTripletLoss.py │ │ │ ├── SpladeLoss.py │ │ │ └── __init__.py │ │ ├── model_card.py │ │ ├── model_card_template.md │ │ ├── models/ │ │ │ ├── MLMTransformer.py │ │ │ ├── SparseAutoEncoder.py │ │ │ ├── SparseStaticEmbedding.py │ │ │ ├── SpladePooling.py │ │ │ └── __init__.py │ │ ├── search_engines.py │ │ ├── trainer.py │ │ └── training_args.py │ ├── trainer.py │ ├── training_args.py │ └── util/ │ ├── __init__.py │ ├── decorators.py │ ├── distributed.py │ ├── environment.py │ ├── file_io.py │ ├── hard_negatives.py │ ├── misc.py │ ├── retrieval.py │ ├── similarity.py │ └── tensor.py └── tests/ ├── __init__.py ├── conftest.py ├── cross_encoder/ │ ├── __init__.py │ ├── conftest.py │ ├── test_backends.py │ ├── test_cross_encoder.py │ ├── test_deprecated_imports.py │ ├── test_model_card.py │ ├── test_multi_process.py │ ├── test_pretrained.py │ ├── test_train_stsb.py │ └── test_trainer.py ├── evaluation/ │ ├── test_binary_classification_evaluator.py │ ├── test_information_retrieval_evaluator.py │ ├── test_label_accuracy_evaluator.py │ ├── test_nanobeir_evaluator.py │ ├── test_paraphrase_mining_evaluator.py │ └── test_triplet_evaluator.py ├── losses/ │ └── test_MatryoshkaLoss.py ├── models/ │ ├── __init__.py │ ├── test_dense.py │ ├── test_pooling.py │ ├── test_router.py │ ├── test_static_embedding.py │ └── test_transformer.py ├── samplers/ │ ├── test_group_by_label_batch_sampler.py │ ├── test_no_duplicates_batch_sampler.py │ └── test_round_robin_batch_sampler.py ├── sparse_encoder/ │ ├── __init__.py │ ├── conftest.py │ ├── models/ │ │ ├── __init__.py │ │ ├── test_csr.py │ │ └── test_sparse_static_embedding.py │ ├── test_backends.py │ ├── test_model_card.py │ ├── test_multi_process.py │ ├── test_opensearch_models.py │ ├── test_pretrained.py │ ├── test_sparse_encoder.py │ ├── test_train_stsb.py │ ├── test_trainer.py │ └── utils.py ├── test_backends.py ├── test_cmnrl.py ├── test_compute_embeddings.py ├── test_custom_models.py ├── test_image_embeddings.py ├── test_model_card.py ├── test_model_card_data.py ├── test_multi_process.py ├── test_pretrained.py ├── test_pretrained_stsb.py ├── test_sentence_transformer.py ├── test_train_stsb.py ├── test_trainer.py ├── test_training_args.py ├── util/ │ ├── test_hard_negatives.py │ ├── test_import.py │ ├── test_retrieval.py │ ├── test_similarity.py │ └── test_tensor.py └── utils.py