psinger commited on
Commit
6c718c9
1 Parent(s): 6894f7c

Update cfg.yaml

Browse files
Files changed (1) hide show
  1. cfg.yaml +2 -4
cfg.yaml CHANGED
@@ -24,7 +24,7 @@ dataset:
24
  text_answer_separator: <|answer|>
25
  text_prompt_start: <|prompt|>
26
  train_dataframe: data/user/oasst/train_full_allrank.pq
27
- validation_dataframe: data/user/oasst/gpt4_val_v0.csv
28
  validation_size: 0.01
29
  validation_strategy: custom
30
  environment:
@@ -43,14 +43,12 @@ environment:
43
  experiment_name: h2ogpt-gm-oasst1-en-2048-falcon-40b-v1
44
  llm_backbone: tiiuae/falcon-40b
45
  logging:
46
- logger: Neptune
47
- neptune_project: Zoo/h2o-llm
48
  number_of_texts: 10
49
  output_directory: output/user/h2ogpt-gm-oasst1-en-2048-falcon-40b-v1/
50
  prediction:
51
  batch_size_inference: 4
52
  do_sample: false
53
- max_length_inference: 512
54
  metric: GPT3.5
55
  min_length_inference: 2
56
  num_beams: 1
 
24
  text_answer_separator: <|answer|>
25
  text_prompt_start: <|prompt|>
26
  train_dataframe: data/user/oasst/train_full_allrank.pq
27
+ validation_dataframe: data/user/oasst/val.csv
28
  validation_size: 0.01
29
  validation_strategy: custom
30
  environment:
 
43
  experiment_name: h2ogpt-gm-oasst1-en-2048-falcon-40b-v1
44
  llm_backbone: tiiuae/falcon-40b
45
  logging:
 
 
46
  number_of_texts: 10
47
  output_directory: output/user/h2ogpt-gm-oasst1-en-2048-falcon-40b-v1/
48
  prediction:
49
  batch_size_inference: 4
50
  do_sample: false
51
+ max_length_inference: 1024
52
  metric: GPT3.5
53
  min_length_inference: 2
54
  num_beams: 1