|
{ |
|
"model_type": "gpt2", |
|
"model_name": "RabbitRedux", |
|
"version": "1.0", |
|
"author": "Canstralian", |
|
"description": "This configuration file defines the model settings for RabbitRedux.", |
|
"vocab_size": 50257, |
|
"hidden_size": 768, |
|
"num_hidden_layers": 12, |
|
"num_attention_heads": 12, |
|
"intermediate_size": 3072, |
|
"hidden_act": "gelu", |
|
"layer_norm_epsilon": 1e-5, |
|
"initializer_range": 0.02, |
|
"dropout": 0.1, |
|
"attention_probs_dropout_prob": 0.1, |
|
"pad_token_id": 50256, |
|
"eos_token_id": 50256, |
|
"bos_token_id": 50256, |
|
"training_params": { |
|
"batch_size": 32, |
|
"learning_rate": 0.001, |
|
"epochs": 10 |
|
}, |
|
"inference_params": { |
|
"max_length": 128, |
|
"temperature": 0.7 |
|
} |
|
} |