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 get_eos_id(&self) -> Result<i64, RustBertError>

pub fn get_tokenizer(&self) -> &TokenizerOption

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>
) -> Vec<Vec<i64>>

Interface method to generate_from_ids_and_past() of the particular models.