Views
No views yet
0.4.01base_model: NousResearch/Meta-Llama-3-70B
2model_type: LlamaForCausalLM
3tokenizer_type: PreTrainedTokenizerFast
4
5#overrides_of_model_config:
6# rope_scaling:
7# type: linear
8# factor: 4
9
10special_tokens:
11 pad_token: "<|end_of_text|>"
12
13gptq: false
14gptq_disable_exllama: true
15
16load_in_8bit: false
17load_in_4bit: true
18strict: false
19
20datasets:
21 - path: /workspace/axolotl/output.jsonl
22 ds_type: json
23 type: completion
24 data_files:
25 - /workspace/axolotl/output.jsonl
26
27output_dir: ./2-qlora-out-l3-10
28
29adapter: qlora
30lora_model_dir:
31
32sequence_len: 2048
33sample_packing: true
34eval_sample_packing: true
35pad_to_sequence_len: true
36
37lora_r: 32
38lora_alpha: 90
39lora_dropout: 0.10
40lora_target_linear: true
41lora_target_modules:
42 - gate_proj
43 - down_proj
44 - up_proj
45 - q_proj
46 - v_proj
47 - k_proj
48 - o_proj
49peft_use_dora: true
50
51wandb_project: kalomaze-model
52wandb_entity:
53wandb_watch:
54wandb_name:
55wandb_log_model:
56
57gradient_accumulation_steps: 1
58micro_batch_size: 2
59num_epochs: 4
60# optimizer: paged_adamw_8bit
61# optimizer: adamw_bnb_8bit
62optimizer: adamw_bnb_8bit
63lr_scheduler: cosine
64learning_rate: 0.000015
65cosine_min_lr_ratio: 0.2
66max_grad_norm: 1.0
67
68train_on_inputs: true
69group_by_length: false
70bf16: true
71fp16: false
72tf32: false
73
74gradient_checkpointing: true
75early_stopping_patience:
76resume_from_checkpoint:
77local_rank:
78logging_steps: 1
79xformers_attention:
80flash_attention: true
81
82warmup_steps: 0
83saves_per_epoch: 2
84save_total_limit: 7
85debug:
86weight_decay: 0.0
87# fsdp:
88# - full_shard
89# - auto_wrap
90# fsdp_config:
91# fsdp_limit_all_gathers: true
92# fsdp_sync_module_states: true
93# fsdp_offload_params: false
94# fsdp_use_orig_params: false
95# fsdp_cpu_ram_efficient_loading: false
96# fsdp_auto_wrap_policy: TRANSFORMER_BASED_WRAP
97# fsdp_transformer_layer_cls_to_wrap: LlamaDecoderLayer
98# fsdp_state_dict_type: FULL_STATE_DICT
99
100seed: 246