|
{ |
|
"architectures": [ |
|
"Custom_MPTForCausalLM" |
|
], |
|
"attn_config": { |
|
"alibi": true, |
|
"alibi_bias_max": 8, |
|
"attn_impl": "torch", |
|
"attn_pdrop": 0, |
|
"attn_type": "multihead_attention", |
|
"attn_uses_sequence_id": false, |
|
"clip_qkv": null, |
|
"prefix_lm": false, |
|
"qk_gn": false, |
|
"qk_ln": false, |
|
"rope": false, |
|
"rope_dail_config": { |
|
"pos_idx_in_fp32": true, |
|
"type": "original", |
|
"xpos_scale_base": 512 |
|
}, |
|
"rope_hf_config": { |
|
"factor": 1.0, |
|
"type": "no_scaling" |
|
}, |
|
"rope_impl": "dail", |
|
"rope_theta": 10000, |
|
"sliding_window_size": -1, |
|
"softmax_scale": null |
|
}, |
|
"auto_map": { |
|
"AutoConfig": "mpt-7b--configuration_mpt.MPTConfig", |
|
"AutoModelForCausalLM": "mpt-7b--modeling_mpt.MPTForCausalLM" |
|
}, |
|
"bos_token_id": 0, |
|
"d_model": 360, |
|
"emb_pdrop": 0, |
|
"embedding_fraction": 1.0, |
|
"eos_token_id": 1, |
|
"expansion_ratio": 5, |
|
"fc_type": "torch", |
|
"ffn_config": { |
|
"fc_type": "torch", |
|
"ffn_type": "mptmlp" |
|
}, |
|
"init_config": { |
|
"emb_init_std": null, |
|
"emb_init_uniform_lim": null, |
|
"fan_mode": "fan_in", |
|
"init_div_is_residual": true, |
|
"init_gain": 0, |
|
"init_nonlinearity": "relu", |
|
"init_std": 0.02, |
|
"name": "kaiming_normal_", |
|
"verbose": 0 |
|
}, |
|
"init_device": "cuda", |
|
"learned_pos_emb": false, |
|
"logit_scale": null, |
|
"max_seq_len": 600, |
|
"model_type": "mpt", |
|
"n_heads": 36, |
|
"n_layers": 36, |
|
"no_bias": true, |
|
"norm_type": "low_precision_layernorm", |
|
"resid_pdrop": 0, |
|
"torch_dtype": "bfloat16", |
|
"transformers_version": "4.35.0", |
|
"use_cache": false, |
|
"use_pad_tok_in_ffn": true, |
|
"verbose": 0, |
|
"vocab_size": 63740 |
|
} |
|
|