1base_model: HuggingFaceTB/cosmo-1b
2model_type: LlamaForCausalLM
3tokenizer_type: LlamaTokenizer
4
5load_in_8bit: false
6load_in_4bit: false
7strict: false
8
9datasets:
10 - path: vicgalle/alpaca-gpt4
11 type: alpaca
12dataset_prepared_path:
13val_set_size: 0.05
14output_dir: ./lisa-out
15
16sequence_len: 2048
17sample_packing: true
18pad_to_sequence_len: true
19
20adapter:
21lora_model_dir:
22lora_r:
23lora_alpha:
24lora_dropout:
25lora_target_linear:
26lora_fan_in_fan_out:
27
28lisa_n_layers: 8
29lisa_step_interval: 10
30lisa_layers_attribute: model.layers
31
32wandb_project: CosmoAlpacaLisa-1b-v0.1
33wandb_entity:
34wandb_watch:
35wandb_name:
36wandb_log_model:
37
38gradient_accumulation_steps: 4
39micro_batch_size: 2
40num_epochs: 1
41optimizer: adamw_bnb_8bit
42lr_scheduler: cosine
43learning_rate: 5e-5
44
45train_on_inputs: false
46group_by_length: false
47bf16: auto
48fp16:
49tf32: false
50
51gradient_checkpointing: true
52early_stopping_patience:
53resume_from_checkpoint:
54local_rank:
55logging_steps: 1
56xformers_attention:
57flash_attention: true
58
59warmup_steps: 10
60evals_per_epoch: 4
61saves_per_epoch: 1
62debug:
63deepspeed:
64weight_decay: 0.0
65fsdp:
66fsdp_config:
67special_tokens:
68