Enum SummarizationOption

pub enum SummarizationOption {
    Bart(BartGenerator),
    T5(T5Generator),
    LongT5(LongT5Generator),
    ProphetNet(ProphetNetConditionalGenerator),
    Pegasus(PegasusConditionalGenerator),
}

Abstraction that holds one particular summarization model, for any of the supported models

Variants

Bart(BartGenerator)

Summarizer based on BART model

T5(T5Generator)

Summarizer based on T5 model

LongT5(LongT5Generator)

Summarizer based on LongT5 model

ProphetNet(ProphetNetConditionalGenerator)

Summarizer based on ProphetNet model

Pegasus(PegasusConditionalGenerator)

Summarizer based on Pegasus model

Implementations

impl SummarizationOption

pub fn new(config: SummarizationConfig) -> Result<Self, RustBertError>

pub fn new_with_tokenizer( config: SummarizationConfig, tokenizer: TokenizerOption, ) -> Result<Self, RustBertError>

pub fn model_type(&self) -> ModelType

Returns the ModelType for this SummarizationOption

pub fn get_tokenizer(&self) -> &TokenizerOption

Interface method to access tokenizer

pub fn get_tokenizer_mut(&mut self) -> &mut TokenizerOption

Interface method to access tokenizer

pub fn generate<S>( &self, prompt_texts: Option<&[S]>, ) -> Result<Vec<String>, RustBertError>

where S: AsRef<str> + Send + Sync,

Interface method to generate() of the particular models.