Views
No views yet
0.4.11base_model: /home/ubuntu/North-Texas-FileSystem/raw_model
2tokenizer_type: AutoTokenizer
3
4load_in_4bit: true
5strict: false
6
7datasets:
8 # huggingface repo
9 # - path: VoyagerYuan/tomato_6k
10 # type: alpaca
11 - path: /home/ubuntu/North-Texas-FileSystem/axolotl/fanqie_6000.jsonl
12 ds_type: json
13 data_files: /home/ubuntu/North-Texas-FileSystem/axolotl/fanqie_6000.jsonl
14 type: alpaca
15# dataset_prepared_path: last_run_prepared
16val_set_size: 0.0
17output_dir: ./outputs/out/qlora-llama3_1-405b_20240821_02
18save_safetensors: true
19
20adapter: qlora
21
22sequence_len: 5000
23sample_packing: true
24pad_to_sequence_len: true
25
26lora_r: 8
27lora_alpha: 16
28lora_dropout: 0.05
29lora_target_modules:
30lora_target_linear: true
31
32gradient_accumulation_steps: 4
33micro_batch_size: 1
34num_epochs: 1
35optimizer: adamw_torch
36lr_scheduler: cosine
37learning_rate: 0.000008
38
39train_on_inputs: false
40group_by_length: false
41bf16: true
42tf32: true
43
44gradient_checkpointing: true
45gradient_checkpointing_kwargs:
46 use_reentrant: true
47logging_steps: 1
48flash_attention: true
49
50warmup_steps: 10
51evals_per_epoch: 4
52saves_per_epoch: 1
53weight_decay: 0.0
54fsdp:
55 - full_shard
56 - auto_wrap
57fsdp_config:
58 fsdp_limit_all_gathers: true
59 fsdp_sync_module_states: true
60 fsdp_offload_params: true
61 fsdp_use_orig_params: false
62 fsdp_cpu_ram_efficient_loading: true
63 fsdp_auto_wrap_policy: TRANSFORMER_BASED_WRAP
64 fsdp_transformer_layer_cls_to_wrap: LlamaDecoderLayer
65 fsdp_state_dict_type: FULL_STATE_DICT
66 fsdp_sharding_strategy: FULL_SHARD
67special_tokens:
68 pad_token: <|finetune_right_pad_id|>
69