Views
No views yet
0.9.01## model
2base_model: SillyTilly/ServiceNow-AI-Apriel-Nemotron-15b-Thinker-Chatml
3model_type: AutoModelForCausalLM
4tokenizer_type: AutoTokenizer
5
6## qlora COPE
7load_in_8bit: false
8load_in_4bit: false
9strict: false
10
11## data
12datasets:
13datasets:
14 - path: Delta-Vector/Ursa-Erebus-16K
15 type: completion
16 field: body
17 - path: Delta-Vector/Ursa-Books-Light-Novels-V1
18 type: completion
19 field: text
20 - path: NewEden/Orion-LIT
21 type: completion
22 field: text
23 - path: Delta-Vector/Ursa-Asstr-V2-18k
24 type: completion
25 field: content
26 - path: Delta-Vector/Ursa-Books-V2
27 type: completion
28 field: text
29 - path: Delta-Vector/Ursa-Scribblehub-7k
30 type: completion
31 field: text
32# - path: Delta-Vector/Ursa-SCP-wiki-1.9K
33# type: completion
34# field: text
35 - path: Delta-Vector/Ursa-Orion-EA-Comp-Filtered
36 type: completion
37 field: Text
38 - path: Delta-Vector/Ursa-HoneyFeed
39 type: completion
40 field: text
41 - path: Delta-Vector/Ursa-Falling-through-the-world
42 type: completion
43 field: content
44shuffle_merged_datasets: true
45dataset_prepared_path: dataset_preparedss
46val_set_size: 0.0
47output_dir: ./Rae-15B-Pretrain
48
49## Liger + CCE
50plugins:
51 - axolotl.integrations.liger.LigerPlugin
52 - axolotl.integrations.cut_cross_entropy.CutCrossEntropyPlugin
53liger_rope: true
54liger_rms_norm: true
55liger_layer_norm: true
56liger_glu_activation: true
57liger_fused_linear_cross_entropy: false
58cut_cross_entropy: true
59
60## CTX settings
61sequence_len: 16384
62sample_packing: true
63eval_sample_packing: false
64pad_to_sequence_len: true
65
66## max grad norm
67max_grad_norm: 0.001
68
69
70## WandB
71wandb_project: Rae
72wandb_entity:
73wandb_watch:
74wandb_name: Pretrain-15B
75wandb_log_model:
76
77## evals
78#evals_per_epoch: 0
79#eval_table_size:
80#eval_max_new_tokens: 128
81
82## hoe params
83gradient_accumulation_steps: 2
84micro_batch_size: 2
85num_epochs: 1
86optimizer: paged_adamw_8bit
87lr_scheduler: cosine
88learning_rate: 1e-5
89
90train_on_inputs: false
91group_by_length: false
92bf16: auto
93fp16:
94tf32: false
95
96gradient_checkpointing: true
97early_stopping_patience:
98resume_from_checkpoint:
99local_rank:
100logging_steps: 1
101xformers_attention:
102flash_attention: true
103s2_attention:
104
105warmup_steps: 50
106saves_per_epoch: 2
107debug:
108deepspeed: ./deepspeed_configs/zero3_bf16.json
109weight_decay: 0.0001
110fsdp:
111fsdp_config:
112special_tokens:
113 pad_token: <pad>
114