Enum rust_bert::pipelines::conversation::ConversationOption

pub enum ConversationOption {
    GPT2(GPT2Generator),
}

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

Variants

GPT2(GPT2Generator)

Conversation based on GPT2 model

Implementations

impl ConversationOption

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

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

pub fn get_eos_id(&self) -> Result<i64, RustBertError>

pub fn get_tokenizer(&self) -> &TokenizerOption

Get a reference to the model tokenizer.

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

Get a mutable reference to the model tokenizer.

pub fn model_type(&self) -> ModelType

Returns the ModelType for this ConversationOption

pub fn generate_from_ids_and_past( &self, input_ids: Tensor, attention_mask: Option<Tensor> ) -> Result<Vec<Vec<i64>>, RustBertError>

Interface method to generate_from_ids_and_past() of the particular models.