Update cfg.yaml
Browse files
cfg.yaml
CHANGED
@@ -9,18 +9,19 @@ augmentation:
|
|
9 |
random_parent_probability: 0.0
|
10 |
skip_parent_probability: 0.0
|
11 |
token_mask_probability: 0.0
|
12 |
-
chatbot_author: H2O.ai
|
13 |
-
chatbot_name: h2oGPT
|
14 |
dataset:
|
15 |
add_eos_token_to_answer: true
|
16 |
add_eos_token_to_prompt: true
|
17 |
answer_column: output
|
|
|
|
|
18 |
data_sample: 1.0
|
19 |
data_sample_choice:
|
20 |
- Train
|
21 |
- Validation
|
22 |
mask_prompt_labels: true
|
23 |
parent_id_column: parent_id
|
|
|
24 |
prompt_column:
|
25 |
- instruction
|
26 |
text_answer_separator: <|answer|>
|
@@ -48,7 +49,6 @@ logging:
|
|
48 |
neptune_project: Zoo/h2o-llm
|
49 |
number_of_texts: 10
|
50 |
output_directory: output/user/h2ogpt-gm-oasst1-en-2048-falcon-7b-v3/
|
51 |
-
personalize: true
|
52 |
prediction:
|
53 |
batch_size_inference: 0
|
54 |
do_sample: false
|
|
|
9 |
random_parent_probability: 0.0
|
10 |
skip_parent_probability: 0.0
|
11 |
token_mask_probability: 0.0
|
|
|
|
|
12 |
dataset:
|
13 |
add_eos_token_to_answer: true
|
14 |
add_eos_token_to_prompt: true
|
15 |
answer_column: output
|
16 |
+
chatbot_author: H2O.ai
|
17 |
+
chatbot_name: h2oGPT
|
18 |
data_sample: 1.0
|
19 |
data_sample_choice:
|
20 |
- Train
|
21 |
- Validation
|
22 |
mask_prompt_labels: true
|
23 |
parent_id_column: parent_id
|
24 |
+
personalize: true
|
25 |
prompt_column:
|
26 |
- instruction
|
27 |
text_answer_separator: <|answer|>
|
|
|
49 |
neptune_project: Zoo/h2o-llm
|
50 |
number_of_texts: 10
|
51 |
output_directory: output/user/h2ogpt-gm-oasst1-en-2048-falcon-7b-v3/
|
|
|
52 |
prediction:
|
53 |
batch_size_inference: 0
|
54 |
do_sample: false
|