Struct rust_bert::pipelines::sentence_embeddings::SentenceEmbeddingsConfig

pub struct SentenceEmbeddingsConfig {
    pub modules_config_resource: Box<dyn ResourceProvider + Send>,
    pub transformer_type: ModelType,
    pub transformer_config_resource: Box<dyn ResourceProvider + Send>,
    pub transformer_weights_resource: Box<dyn ResourceProvider + Send>,
    pub pooling_config_resource: Box<dyn ResourceProvider + Send>,
    pub dense_config_resource: Option<Box<dyn ResourceProvider + Send>>,
    pub dense_weights_resource: Option<Box<dyn ResourceProvider + Send>>,
    pub sentence_bert_config_resource: Box<dyn ResourceProvider + Send>,
    pub tokenizer_config_resource: Box<dyn ResourceProvider + Send>,
    pub tokenizer_vocab_resource: Box<dyn ResourceProvider + Send>,
    pub tokenizer_merges_resource: Option<Box<dyn ResourceProvider + Send>>,
    pub device: Device,
}

Configuration for sentence embeddings

Contains information regarding the transformer model to load, the optional extra layers, and device to place the model on.

Fields

modules_config_resource: Box<dyn ResourceProvider + Send>

Modules configuration resource, contains layers definition

transformer_type: ModelType

Transformer model type

transformer_config_resource: Box<dyn ResourceProvider + Send>

Transformer model configuration resource

transformer_weights_resource: Box<dyn ResourceProvider + Send>

Transformer weights resource

pooling_config_resource: Box<dyn ResourceProvider + Send>

Pooling layer configuration resource

dense_config_resource: Option<Box<dyn ResourceProvider + Send>>

Optional dense layer configuration resource

dense_weights_resource: Option<Box<dyn ResourceProvider + Send>>

Optional dense layer weights resource

sentence_bert_config_resource: Box<dyn ResourceProvider + Send>

Sentence BERT specific configuration resource

tokenizer_config_resource: Box<dyn ResourceProvider + Send>

Transformer’s tokenizer configuration resource

tokenizer_vocab_resource: Box<dyn ResourceProvider + Send>

Transformer’s tokenizer vocab resource

tokenizer_merges_resource: Option<Box<dyn ResourceProvider + Send>>

Optional transformer’s tokenizer merges resource

device: Device

Device to place the transformer model on

Trait Implementations

impl From<SentenceEmbeddingsModelType> for SentenceEmbeddingsConfig

fn from(model_type: SentenceEmbeddingsModelType) -> Self

Converts to this type from the input type.