{ "_bertblocks_attn_backend": "sdpa", "_unpadding": false, "actv_fn": "gelu", "add_timestep_emb": false, "add_token_type_emb": false, "architectures": [ "BertBlocksForMaskedLM" ], "attention_gate": null, "attn_dropout_prob": 0.1, "attn_out_bias": false, "attn_proj_bias": false, "block_pos_enc_kind": "alibi-structured-v3", "block_pos_enc_kwargs": { "asym_init_diversity": "linspace", "gate_alpha_lr_multiplier": 50.0, "max_seq_len": 1024, "rope_dim": 64, "slope_init": "alibi" }, "bos_token_id": 50281, "classifier_dropout": 0.1, "dtype": "float32", "emb_dropout_prob": 0.0, "emb_pos_enc_kind": "none", "emb_pos_enc_kwargs": {}, "freeze_pos_enc": false, "global_attention_every_n_layers": 0, "head_type": "proj", "hidden_dropout_prob": 0.0, "hidden_size": 768, "include_final_norm": true, "initializer_cutoff_factor": 3.0, "initializer_gain": 1.0, "initializer_kind": "trunc_normal", "initializer_range": 0.02, "intermediate_size": 1152, "local_attention": [ -1, -1 ], "mask_token_id": 1, "max_sequence_length": 1024, "mlp_in_bias": false, "mlp_out_bias": false, "mlp_type": "glu", "model_type": "bertblocks", "norm_bias": false, "norm_eps": 1e-06, "norm_fn": "layer", "norm_kind": "pre", "norm_params": {}, "norm_qk": false, "norm_scaling": false, "num_attention_heads": 12, "num_blocks": 22, "num_classes": 2, "num_kv_heads": 12, "pad_token_id": 50283, "problem_type": "regression", "relative_attention_max_distance": 128, "relative_attention_num_buckets": 32, "residual_first_layer": false, "transformers_version": "4.57.6", "type_vocab_size": 1, "vocab_size": 50368 }