trl-internal-testing/tiny-BloomForCausalLM

1{2  "apply_residual_connection_post_layernorm": false,3  "architectures": [4    "BloomForCausalLM"5  ],6  "attention_dropout": 0.0,7  "bos_token_id": 1,8  "dtype": "float32",9  "eos_token_id": 2,10  "hidden_dropout": 0.0,11  "hidden_size": 8,12  "initializer_range": 0.02,13  "intermediate_size": 32,14  "layer_norm_epsilon": 1e-05,15  "model_type": "bloom",16  "n_head": 4,17  "n_layer": 2,18  "num_key_value_heads": 2,19  "pretraining_tp": 1,20  "slow_but_exact": false,21  "transformers_version": "4.57.0.dev0",22  "use_cache": true,23  "vocab_size": 25068024}25