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)

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

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.