| { | |
| "num_layers": 80, | |
| "hidden_size": 8192, | |
| "ffn_hidden_size": 28672, | |
| "layernorm_epsilon": 1e-05, | |
| "max_sequence_length": 2048, | |
| "num_attention_heads": 64, | |
| "num_key_value_heads": 8, | |
| "max_position_embeddings": 4096, | |
| "tensor_model_parallel_size": 1, | |
| "pipeline_model_parallel_size": 1, | |
| "data_parallel_size": 1, | |
| "use_distributed_optimizer": true, | |
| "use_rotary_position_embeddings": true, | |
| "add_embedding_padding": true, | |
| "make_vocab_size_divisible_by": 128, | |
| "vocab_size": 32000, | |
| "transpose_mlp_dense": true, | |
| "transpose_query_key_value": true, | |
| "load_platform": "megatron", | |
| "save_platform": "huggingface", | |
| "load_ckpt_path": "/mnt/pfs/jinfeng_team/MMGroup/YWH/LLM_model_cache/exp_checkpoint/mumath_code_stage_2_pot_bs8/iter_0025128", | |
| "save_ckpt_path": "/mnt/pfs/jinfeng_team/MMGroup/YWH/LLM_model_cache/exp_checkpoint/mumath_code_stage_2_pot_bs8/iter_0025128_hf", | |
| "common_config_path": "config/llama-2-70b.json", | |
| "no_load_optim": true, | |
| "no_save_optim": false, | |
| "megatron_path": "/workspace/Megatron-LM/", | |
| "_name_or_path": "meta-llama/Llama-2-70b-hf", | |
| "architectures": [ | |
| "LlamaForCausalLM" | |
| ], | |
| "bos_token_id": 1, | |
| "eos_token_id": 2, | |
| "hidden_act": "silu", | |
| "intermediate_size": 28672, | |
| "initializer_range": 0.02, | |
| "model_type": "llama", | |
| "num_hidden_layers": 80, | |
| "pad_token_id": 0, | |
| "rms_norm_eps": 1e-05, | |
| "tie_word_embeddings": false, | |
| "pretraining_tp": 1, | |
| "use_cache": true | |
| } |