Enum rust_bert::pipelines::translation::TranslationOption

pub enum TranslationOption {
    Marian(MarianGenerator),
    T5(T5Generator),
}

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

Variants

Marian(MarianGenerator)

Translator based on Marian model

T5(T5Generator)

Translator based on T5 model

Implementations

impl TranslationOption

pub fn new(config: TranslationConfig) -> Self

pub fn model_type(&self) -> ModelType

Returns the ModelType for this TranslationOption

pub fn generate(
    &self,
    prompt_texts: Option<Vec<&str>>,
    attention_mask: Option<Tensor>
) -> Vec<String>

Interface method to generate() of the particular models.