NeMo
English
nvidia
rlhf
llama3
zhilinw commited on
Commit
9caa169
1 Parent(s): e5873af

Update model_config.yaml

Browse files
Files changed (1) hide show
  1. model_config.yaml +3 -8
model_config.yaml CHANGED
@@ -40,11 +40,6 @@ num_query_groups: 8
40
  tokenizer:
41
  library: huggingface
42
  type: meta-llama/Meta-Llama-3-70B
43
- model: /dataset/models/llama2-13b/llama-tokenizer.model
44
- vocab_file: null
45
- merge_file: null
46
- tokenizer_model: /dataset/models/llama2-13b/llama-tokenizer.model
47
- sentencepiece_legacy: false
48
  use_fast: true
49
  native_amp_init_scale: 4294967296
50
  native_amp_growth_interval: 1000
@@ -227,11 +222,11 @@ data:
227
  index_mapping_dir: null
228
  data_prefix:
229
  train:
230
- - /lustre/fsw/coreai_dlalgo_llm/geshen/trt_llm/helpsteer_data/helpsteer_11_train_prompts.jsonl
231
  validation:
232
- - /lustre/fsw/coreai_dlalgo_llm/geshen/trt_llm/helpsteer_data/helpsteer_11_val_prompts.jsonl
233
  test:
234
- - /lustre/fsw/coreai_dlalgo_llm/geshen/trt_llm/helpsteer_data/helpsteer_11_val_prompts.jsonl
235
  nsys_profile:
236
  enabled: false
237
  start_step: 10
 
40
  tokenizer:
41
  library: huggingface
42
  type: meta-llama/Meta-Llama-3-70B
 
 
 
 
 
43
  use_fast: true
44
  native_amp_init_scale: 4294967296
45
  native_amp_growth_interval: 1000
 
222
  index_mapping_dir: null
223
  data_prefix:
224
  train:
225
+ - /dataset/train.jsonl
226
  validation:
227
+ - /dataset/val.jsonl
228
  test:
229
+ - /dataset/val.jsonl
230
  nsys_profile:
231
  enabled: false
232
  start_step: 10