Enum rust_bert::pipelines::translation::TranslationOption

pub enum TranslationOption {
    Marian(MarianGenerator),
    T5(T5Generator),
    MBart(MBartGenerator),
    M2M100(M2M100Generator),
}

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

Variants

Marian(MarianGenerator)

Tuple Fields

0: MarianGenerator

Translator based on Marian model

T5(T5Generator)

Tuple Fields

0: T5Generator

Translator based on T5 model

MBart(MBartGenerator)

Tuple Fields

0: MBartGenerator

Translator based on MBart50 model

M2M100(M2M100Generator)

Tuple Fields

0: M2M100Generator

Translator based on M2M100 model

Implementations

impl TranslationOption

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

pub fn model_type(&self) -> ModelType

Returns the ModelType for this TranslationOption

pub fn generate<S>(
    &self,
    prompt_texts: Option<&[S]>,
    forced_bos_token_id: Option<i64>
) -> Vec<String> where
    S: AsRef<str> + Sync, 

Interface method to generate() of the particular models.