nort5-small / config.json
davda54's picture
Update config.json
d005f2c
{
"architectures": [
"NorT5ForConditionalGeneration"
],
"auto_map": {
"AutoConfig": "configuration_nort5.NorT5Config",
"AutoModel": "modeling_nort5.NorT5Model",
"AutoModelForSeq2SeqLM": "modeling_nort5.NorT5ForConditionalGeneration",
"AutoModelForConditionalGeneration": "modeling_nort5.NorT5ForConditionalGeneration"
},
"attention_probs_dropout_prob": 0.0,
"bos_token_id": 5,
"cls_token_id": 1,
"eos_token_id": 6,
"hidden_dropout_prob": 0.0,
"hidden_size": 384,
"initializer_range": 0.02,
"intermediate_size": 1024,
"layer_norm_eps": 1e-07,
"max_position_embeddings": 512,
"num_attention_heads": 6,
"num_hidden_layers": 12,
"output_all_encoded_layers": true,
"pad_token_id": 3,
"position_bucket_size": 32,
"sep_token_id": 2,
"torch_dtype": "float32",
"transformers_version": "4.24.0",
"vocab_size": 50000,
"max_length": 512,
"max_new_tokens": 256,
"is_encoder_decoder": true
}