mirror of
https://github.com/guillaume-be/rust-bert.git
synced 2024-08-16 16:10:25 +03:00
* 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 |