File size: 897 Bytes
fe3a312 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
{
"_name_or_path": "/mnt/shared-storage/tenant/opensource/step_llm",
"allow_transformer_engine": false,
"architectures": [
"Step1Model"
],
"attention_dropout": 0.0,
"attention_impl": "GQA",
"base_batch_size": 128,
"embedding_weights_in_fp32": false,
"ffn_hidden_size": 16896,
"fp32_residual_connection": false,
"hidden_dropout": 0.0,
"hidden_size": 6144,
"kv_channels": 128,
"layernorm_epsilon": 1e-05,
"max_position_embeddings": 16384,
"num_attention_groups": 8,
"num_attention_heads": 48,
"num_layers": 48,
"orig_vocab_size": 65536,
"overlap_p2p_comm": true,
"padded_vocab_size": 65536,
"params_dtype": "torch.bfloat16",
"seq_length": 16384,
"swiglu_recompute_silu_dot": true,
"tokens_to_generate": 512,
"torch_dtype": "bfloat16",
"transformers_version": "4.48.3",
"use_flash_attn": true,
"virtual_pipeline_model_parallel_size": 3
}
|