Upload folder using huggingface_hub

config.json

@@ -0,0 +1,26 @@
+{
+  "_name_or_path": "/tmp/llama2.13b.wudao.sft.combine.v1.0.seq2k.w16.adamw.NA100.0803.ds/checkpoint-1750",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 5120,
+  "initializer_range": 0.02,
+  "intermediate_size": 13824,
+  "max_position_embeddings": 2048,
+  "model_type": "llama",
+  "num_attention_heads": 40,
+  "num_hidden_layers": 40,
+  "num_key_value_heads": 40,
+  "pad_token_id": 0,
+  "pretraining_tp": 2,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.31.0",
+  "use_cache": false,
+  "vocab_size": 79458
+}

generation_config.json

@@ -0,0 +1,9 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "pad_token_id": 0,
+  "temperature": 0.9,
+  "top_p": 0.6,
+  "transformers_version": "4.31.0"
+}

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ec8bca7e83bce757be0c5c86dde381c7f6c069c288da15d17522041833b9ff65
+size 27003737780

special_tokens_map.json

@@ -0,0 +1,24 @@
+{
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": "<unk>",
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer.model

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3c6776001191a59ce6be0d949adb3b524f37fc1fbd387e48d2ab7b1310912bdf
+size 1268131

tokenizer_config.json

@@ -0,0 +1,32 @@
+{
+  "bos_token": {
+    "__type": "AddedToken",
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "clean_up_tokenization_spaces": false,
+  "eos_token": {
+    "__type": "AddedToken",
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "legacy": true,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": null,
+  "sp_model_kwargs": {},
+  "tokenizer_class": "LlamaTokenizer",
+  "unk_token": {
+    "__type": "AddedToken",
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  }
+}

training_config.yaml

@@ -0,0 +1,123 @@
+aws_output_bucket: s3://panda-us-west-2/experiments/llama2.13b.wudao.sft.combine.legal.v1.0.seq2k.w16.adamw.NA100.0921.ds
+data_dir: null
+dist_load_data_barrier: false
+train_file: data/files/tmp.json
+dev_file: null
+test_file: null
+model:
+  _target_: models.llama.LlamaForConditionalGeneration.from_pretrained
+  use_peft: false
+  gradient_checkpointing: true
+  enable_flash_attention: true
+  flash_attention_vanilla_torch: true
+read_tensor:
+  _target_: data.collators.zh_instruct.TextDatasetUnifyV3
+  pair_file_list: null
+extended_vocab: null
+collator:
+  _target_: data.collators.flan.FlanCollatorOverCollator
+  collator: null
+  max_seq_length: 2048
+  tokenizer: ${model_name_or_path}
+  decoder_only: true
+  padding: longest
+  padding_side: right
+num_workers: 4
+prefetch_factor: 2
+do_preprocess: false
+model_name_or_path: /tmp/llama2.13b.wudao.sft.combine.v1.0.seq2k.w16.adamw.NA100.0803.ds/checkpoint-1750
+pretrain: null
+exp_name: llama2.13b.wudao.sft.combine.legal.v1.0.seq2k.w16.adamw.NA100.0921.ds
+exp_notes: null
+output_dir: /tmp/${exp_name}
+resume: null
+do_train: true
+evaluate_during_training: false
+do_eval: false
+eval_sub_path: checkpoint-*
+per_gpu_train_batch_size: 2
+per_gpu_eval_batch_size: 1
+learning_rate: 1.0e-06
+gradient_accumulation_steps: 8
+weight_decay: 0.01
+adam_epsilon: 1.0e-06
+adam_betas: (0.9, 0.99)
+max_grad_norm: 1.0
+num_train_epochs: 1
+total_dataset_len: -1
+max_steps: 0
+warmup_proportion: 0
+warmup_steps: 0
+optimizer: null
+use_nvlamb: null
+bit_training: null
+logging_steps: 1
+save_best: false
+save_steps: 250
+eval_steps: 250
+ddp_eval: true
+no_cuda: false
+seed: 42
+local_rank: 0
+fp16: true
+fp16_opt_level: O1
+fp16_bfloat16: true
+prediction_cfg:
+  metric: acc
+  measure: 1
+  best_checkpoint: null
+  best_result: null
+eval_forward_fn:
+  _target_: general_util.evaluator.DiscriminatorForwardFn
+post_process: null
+fairscale_config:
+  _target_: general_util.fsdp_utils.default_initialize
+  fp16: ${fp16}
+  move_grads_to_cpu: false
+  move_params_to_cpu: false
+  flatten_parameters: false
+with_lightseq: false
+load_lr_scheduler_states: false
+ds_cfg:
+  train_micro_batch_size_per_gpu: ${per_gpu_train_batch_size}
+  gradient_accumulation_steps: ${gradient_accumulation_steps}
+  optimizer:
+    type: AdamW
+    params:
+      lr: ${learning_rate}
+      betas:
+      - 0.9
+      - 0.96
+      eps: ${adam_epsilon}
+      weight_decay: ${weight_decay}
+  scheduler:
+    type: WarmupDecayLR
+    params:
+      total_num_steps: 1474
+      warmup_max_lr: ${learning_rate}
+      warmup_num_steps: 0
+      warmup_type: linear
+  gradient_clipping: ${max_grad_norm}
+  bf16:
+    enabled: ${fp16}
+  zero_optimization:
+    stage: 1
+    contiguous_gradients: true
+    overlap_comm: true
+    reduce_scatter: true
+    reduce_bucket_size: 500000000.0
+    allgather_bucket_size: 500000000.0
+    offload_optimizer:
+      device: cpu
+      pin_memory: true
+  steps_per_print: 1
+summary_helper:
+  _target_: general_util.tensorboard_helper.WandbWriter
+  batch_index_or_keys: null
+  outputs_index_or_keys: null
+n_gpu: 1
+device: cuda:0
+train_batch_size: 2
+eval_batch_size: null
+world_size: 16
+world_rank: null