Enum rust_bert::pipelines::translation::TranslationOption

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

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

MBart(MBartGenerator)

Translator based on MBart50 model

M2M100(M2M100Generator)

Translator based on M2M100 model

NLLB(NLLBGenerator)

Translator based on NLLB model

Implementations

impl TranslationOption

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

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

pub fn model_type(&self) -> ModelType

Returns the ModelType for this TranslationOption

pub fn get_tokenizer(&self) -> &TokenizerOption

Returns the Tokenizer for this TranslationOption

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

Interface method to access tokenizer

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

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

Interface method to generate() of the particular models.