1base_model: Qwen/Qwen1.5-0.5B
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4trust_remote_code: true
5
6
7load_in_8bit: false
8load_in_4bit: false
9strict: false
10
11datasets:
12 - path: silk-road/ChatHaruhi-RolePlaying
13 type: "completion"
14dataset_prepared_path:
15val_set_size: 0.00
16output_dir: ./outputs/out
17
18sequence_len: 250
19sample_packing: true
20pad_to_sequence_len: true
21save_safetensors: true
22gpu_memory_limit: 80GiB
23
24adapter:
25lora_model_dir:
26lora_r:
27lora_alpha:
28lora_dropout:
29lora_target_linear:
30lora_fan_in_fan_out:
31
32wandb_project:
33wandb_entity:
34wandb_watch:
35wandb_name:
36wandb_log_model:
37
38gradient_accumulation_steps: 1
39micro_batch_size: 4
40num_epochs: 1
41optimizer: adamw_bnb_8bit
42lr_scheduler: cosine
43learning_rate: 0.00005
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: false #toggle
58
59warmup_steps: 100
60evals_per_epoch: 4
61eval_table_size:
62saves_per_epoch: 1
63debug:
64deepspeed: #deepspeed_configs/zero2.json # multi-gpu only
65weight_decay: 0.01
66fsdp:
67fsdp_config:
68special_tokens:
69