pub type PegasusConfig = MBartConfig;Expand description
§Pegasus model configuration
Defines the Pegasus model architecture (e.g. number of layers, hidden layer size, label mapping…)
Aliased Type§
struct PegasusConfig {Show 33 fields
pub vocab_size: i64,
pub max_position_embeddings: i64,
pub encoder_layers: i64,
pub encoder_attention_heads: i64,
pub encoder_ffn_dim: i64,
pub encoder_layerdrop: f64,
pub decoder_layers: i64,
pub decoder_ffn_dim: i64,
pub decoder_attention_heads: i64,
pub decoder_layerdrop: f64,
pub is_encoder_decoder: Option<bool>,
pub activation_function: Option<Activation>,
pub d_model: i64,
pub dropout: f64,
pub activation_dropout: f64,
pub attention_dropout: f64,
pub classifier_dropout: Option<f64>,
pub scale_embedding: Option<bool>,
pub bos_token_id: Option<i64>,
pub eos_token_id: Option<i64>,
pub pad_token_id: Option<i64>,
pub forced_bos_token_id: Option<i64>,
pub forced_eos_token_id: Option<i64>,
pub decoder_start_token_id: Option<i64>,
pub id2label: Option<HashMap<i64, String>>,
pub label2id: Option<HashMap<String, i64>>,
pub init_std: f64,
pub min_length: Option<i64>,
pub no_repeat_ngram_size: Option<i64>,
pub normalize_embedding: Option<bool>,
pub output_attentions: Option<bool>,
pub output_hidden_states: Option<bool>,
pub output_past: Option<bool>,
}Fields§
§vocab_size: i64§max_position_embeddings: i64§encoder_layers: i64§encoder_attention_heads: i64§encoder_ffn_dim: i64§encoder_layerdrop: f64§decoder_layers: i64§decoder_ffn_dim: i64§decoder_attention_heads: i64§decoder_layerdrop: f64§is_encoder_decoder: Option<bool>§activation_function: Option<Activation>§d_model: i64§dropout: f64§activation_dropout: f64§attention_dropout: f64§classifier_dropout: Option<f64>§scale_embedding: Option<bool>§bos_token_id: Option<i64>§eos_token_id: Option<i64>§pad_token_id: Option<i64>§forced_bos_token_id: Option<i64>§forced_eos_token_id: Option<i64>§decoder_start_token_id: Option<i64>§id2label: Option<HashMap<i64, String>>§label2id: Option<HashMap<String, i64>>§init_std: f64§min_length: Option<i64>§no_repeat_ngram_size: Option<i64>§normalize_embedding: Option<bool>§output_attentions: Option<bool>§output_past: Option<bool>