Enum rust_bert::pipelines::summarization::SummarizationOption

pub enum SummarizationOption {
    Bart(BartGenerator),
    T5(T5Generator),
}

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

Implementations

impl SummarizationOption

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

pub fn model_type(&self) -> ModelType

Returns the ModelType for this SummarizationOption

pub fn generate<'a, S>(
    &self,
    prompt_texts: Option<S>,
    attention_mask: Option<Tensor>
) -> Vec<String> where
    S: AsRef<[&'a str]>, 

Interface method to generate() of the particular models.