Views
No views yet
0.4.11base_model: google/gemma-2-9b
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4
5load_in_8bit: true
6load_in_4bit: false
7strict: false
8
9# huggingface repo
10chat_template: gemma
11datasets:
12 - path: cgato/SlimOrcaDedupCleaned
13 type: chat_template
14 chat_template: gemma
15 drop_system_message: true
16val_set_size: 0.0
17output_dir: ./outputs/out
18
19adapter: lora
20lora_r: 32
21lora_alpha: 16
22lora_dropout: 0.05
23lora_target_linear: true
24lora_modules_to_save:
25 - embed_tokens
26 - lm_head
27
28
29sequence_len: 2048
30sample_packing: true
31eval_sample_packing: false
32pad_to_sequence_len: true
33
34wandb_project: gemma2-exp
35wandb_entity: oaaic
36wandb_watch:
37wandb_name:
38wandb_log_model:
39
40
41gradient_accumulation_steps: 4
42micro_batch_size: 4
43num_epochs: 1
44optimizer: adamw_bnb_8bit
45adam_beta2: 0.95
46adam_eps: 0.00001
47max_grad_norm: 1.0
48lr_scheduler: cosine
49learning_rate: 0.00003
50
51train_on_inputs: false
52group_by_length: false
53bf16: auto
54fp16:
55tf32: true
56
57gradient_checkpointing: true
58early_stopping_patience:
59resume_from_checkpoint:
60local_rank:
61logging_steps: 1
62xformers_attention:
63flash_attention: true
64
65warmup_ratio: 0.1
66evals_per_epoch:
67eval_table_size:
68eval_max_new_tokens: 128
69saves_per_epoch: 2
70debug:
71deepspeed:
72weight_decay: 0.1
73fsdp:
74fsdp_config:
75special_tokens:
76