Support uc save for deepseek

llm/config/deepseek-v3/sft_argument.json

@@ -1,19 +1,20 @@
 {
-  "model_name_or_path": "/root/paddlejob/tmpspace/huggingface_model/huggingface/deepseek-ai/DeepSeek-V3-bf16/",
+  "model_name_or_path": "./dsv3_128k_config/",
   "dataset_name_or_path": "./data",
   "output_dir": "./checkpoints/sft_ckpts",
   "per_device_train_batch_size": 1,
   "gradient_accumulation_steps": 16,
   "per_device_eval_batch_size": 1,
   "eval_accumulation_steps": 1,
-  "max_steps": 20,
+  "max_steps": 5,
+  "save_steps": 3,
   "amp_master_grad": true,
   "num_train_epochs": 1,
   "learning_rate": 2.2e-04,
   "warmup_steps": 30,
   "logging_steps": 1,
   "evaluation_strategy": "no",
-  "save_strategy": "no",
+  "save_strategy": "steps",
   "src_length": 2048,
   "max_length": 4097,
   "bf16": true,
@@ -22,32 +23,33 @@
   "do_eval": false,
   "disable_tqdm": true,
   "use_expert_parallel": true,
-  "expert_parallel_degree": 16,
+  "expert_parallel_degree": 4,
   "continue_training": false,
   "pipeline_parallel_config": "enable_delay_scale_loss disable_partial_send_recv disable_batch_p2p_comm",
   "tensor_parallel_config": "enable_delay_scale_loss",
   "sharding_parallel_config": "split_param",
-  "load_best_model_at_end": true,
   "eval_with_do_generation": false,
   "metric_for_best_model": "loss",
   "recompute": true,
   "recompute_use_reentrant": true,
   "recompute_granularity": "full",
   "save_total_limit": 1,
-  "tensor_parallel_degree": 1,
-  "pipeline_parallel_degree": 8,
-  "sharding_parallel_degree": 16,
+  "tensor_parallel_degree": 2,
+  "pipeline_parallel_degree": 2,
+  "sharding_parallel_degree": 2,
   "sharding": "stage1",
   "zero_padding": true,
   "unified_checkpoint": true,
+  "unified_checkpoint_config": "ignore_merge_optimizer",
   "use_flash_attention": true,
   "flash_mask": true,
   "using_fake_gate": true,
   "using_flex_token": true,
   "use_fused_rms_norm": true,
   "moe_subbatch_token_num": 0,
-  "pre_alloc_memory": 60,
+  "pre_alloc_memory": 5,
   "tensorwise_offload_optimizer": true,
-  "sequence_parallel": false,
+  "sequence_parallel": true,
   "tensor_parallel_output": true
 }
+

llm/deepseeek_sft.sh

@@ -17,6 +17,8 @@
 # wget https://bj.bcebos.com/paddlenlp/models/transformers/llama/data/llama_openwebtext_100k.bin
 # wget https://bj.bcebos.com/paddlenlp/models/transformers/llama/data/llama_openwebtext_100k.idx
 
-mpirun rm -rf output
-nohup mpirun sh script/train_gpu.sh config/deepseek-v3/sft_argument.json > run.log 2>&1 & 
+#mpirun rm -rf output
+#nohup mpirun sh script/train_gpu.sh config/deepseek-v3/sft_argument.json > run.log 2>&1 & 
+
+nohup sh script/train_gpu.sh config/deepseek-v3/sft_argument.json > run.log 2>&1 &
 

llm/dsv3_128k_config/config.json

@@ -0,0 +1,61 @@
+{
+    "architectures": [
+      "DeepseekV3ForCausalLM"
+    ],
+    "attention_bias": false,
+    "attention_dropout": 0.0,
+    "auto_map": {
+      "AutoConfig": "configuration_deepseek.DeepseekV3Config",
+      "AutoModel": "modeling_deepseek.DeepseekV3Model",
+      "AutoModelForCausalLM": "modeling_deepseek.DeepseekV3ForCausalLM"
+    },
+    "aux_loss_alpha": 0.001,
+    "bos_token_id": 0,
+    "eos_token_id": 1,
+    "ep_size": 1,
+    "first_k_dense_replace": 3,
+    "hidden_act": "silu",
+    "hidden_size": 7168,
+    "initializer_range": 0.02,
+    "intermediate_size": 18432,
+    "kv_lora_rank": 512,
+    "max_position_embeddings": 163840,
+    "model_type": "deepseek_v3",
+    "moe_intermediate_size": 2048,
+    "moe_layer_freq": 1,
+    "n_group": 8,
+    "n_routed_experts": 16,
+    "n_shared_experts": 1,
+    "norm_topk_prob": true,
+    "num_attention_heads": 128,
+    "num_experts_per_tok": 8,
+    "num_hidden_layers": 7,
+    "num_key_value_heads": 128,
+    "num_nextn_predict_layers": 1,
+    "pretraining_tp": 1,
+    "q_lora_rank": 1536,
+    "qk_nope_head_dim": 128,
+    "qk_rope_head_dim": 64,
+    "rms_norm_eps": 1e-06,
+    "rope_scaling": {
+      "beta_fast": 32,
+      "beta_slow": 1,
+      "factor": 40,
+      "mscale": 1.0,
+      "mscale_all_dim": 1.0,
+      "original_max_position_embeddings": 4096,
+      "type": "yarn"
+    },
+    "rope_theta": 10000,
+    "routed_scaling_factor": 2.5,
+    "scoring_func": "sigmoid",
+    "seq_aux": true,
+    "tie_word_embeddings": false,
+    "topk_group": 4,
+    "topk_method": "noaux_tc",
+    "torch_dtype": "bfloat16",
+    "transformers_version": "4.33.1",
+    "use_cache": true,
+    "v_head_dim": 128,
+    "vocab_size": 129280
+  }

llm/dsv3_128k_config/configuration.json

@@ -0,0 +1 @@
+{"framework": "pytorch", "task": "text-generation", "allow_remote": true}