mirror of
https://github.com/guillaume-be/rust-bert.git
synced 2024-08-16 16:10:25 +03:00
4d8a298586
* Add sbert implementation for inference * Fix clippy warnings * Refactor sentence embeddings into a dedicated pipeline * Add output_attentions and output_hidden_states to T5Config * Add sbert implementation for inference * Fix clippy warnings * Refactor sentence embeddings into a dedicated pipeline * Add output_attentions and output_hidden_states to T5Config * Improve sentence_embeddings implementation * Dedicated tokenizer config for strip_accents and add_prefix_space * Rename forward to encode_as_tensor * Remove _conf from Dense layer * Add sentence embeddings docs * Addition of remote resources and tests update * Merge feature branch and fix doctests * Add SentenceEmbeddingsBuilder<Remote> and improve remote resources * Use tch::no_grad in sentence embeddings * Updated changelog, registration of sentence embeddings integration tests Co-authored-by: Guillaume Becquin <guillaume.becquin@gmail.com> |
||
|---|---|---|
| .. | ||
| async-sentiment.rs | ||
| conversation.rs | ||
| generation_gpt2.rs | ||
| generation_gpt_neo.rs | ||
| generation_reformer.rs | ||
| generation_xlnet.rs | ||
| masked_language_model_bert.rs | ||
| named_entities_recognition.rs | ||
| natural_language_inference_deberta.rs | ||
| part_of_speech_tagging.rs | ||
| question_answering_bert.rs | ||
| question_answering_longformer.rs | ||
| question_answering_squad.rs | ||
| question_answering.rs | ||
| sentence_embeddings_local.rs | ||
| sentence_embeddings.rs | ||
| sentiment_analysis_fnet.rs | ||
| sentiment_analysis_sst2.rs | ||
| sentiment_analysis.rs | ||
| sequence_classification_multilabel.rs | ||
| sequence_classification.rs | ||
| summarization_bart.rs | ||
| summarization_pegasus.rs | ||
| summarization_prophetnet.rs | ||
| summarization_t5.rs | ||
| token_classification.rs | ||
| translation_builder.rs | ||
| translation_m2m100.rs | ||
| translation_marian.rs | ||
| translation_mbart.rs | ||
| translation_t5.rs | ||
| zero_shot_classification.rs | ||