ehartford commited on
Commit
b4421d1
1 Parent(s): b367946

Upload phi-samantha.yml

Browse files
Files changed (1) hide show
  1. axolotl-config/phi-samantha.yml +76 -0
axolotl-config/phi-samantha.yml ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: microsoft/phi-1_5
2
+ base_model_config: microsoft/phi-1_5
3
+ model_type: AutoModelForCausalLM
4
+ tokenizer_type: AutoTokenizer
5
+ is_llama_derived_model: false
6
+ trust_remote_code: true
7
+
8
+ load_in_8bit: false
9
+ load_in_4bit: false
10
+ strict: false
11
+
12
+ datasets:
13
+ - path: ehartford/samantha-data
14
+ type: sharegpt:chat
15
+ data_files: samantha-1.1.json
16
+
17
+ dataset_prepared_path: last_run_prepared
18
+ val_set_size: 0.01
19
+ output_dir: /workspace/samantha-phi
20
+
21
+ sequence_len: 2048
22
+ sample_packing: false
23
+ pad_to_sequence_len:
24
+
25
+ adapter:
26
+ lora_model_dir:
27
+ lora_r:
28
+ lora_alpha:
29
+ lora_dropout:
30
+ lora_target_linear:
31
+ lora_fan_in_fan_out:
32
+
33
+ wandb_project: samantha-phi
34
+ wandb_entity:
35
+ wandb_watch:
36
+ wandb_run_id:
37
+ wandb_log_model:
38
+
39
+ gradient_accumulation_steps: 1
40
+ micro_batch_size: 12
41
+ num_epochs: 2000
42
+ optimizer: adamw_bnb_8bit
43
+ adam_beta2: 0.99
44
+ adam_epsilon: 0.00001
45
+ max_grad_norm: 1.0
46
+ lr_scheduler: cosine
47
+ learning_rate: 0.0002
48
+
49
+ train_on_inputs: false
50
+ group_by_length: true
51
+ bf16: true
52
+ fp16: false
53
+ tf32: true
54
+
55
+ gradient_checkpointing:
56
+ early_stopping_patience:
57
+ resume_from_checkpoint:
58
+ local_rank:
59
+ logging_steps: 1
60
+ xformers_attention: true
61
+ flash_attention:
62
+
63
+ warmup_steps: 100
64
+ eval_steps: 0.05
65
+ save_steps:
66
+ debug:
67
+ deepspeed: deepspeed/zero2.json
68
+ weight_decay: 0.01
69
+ fsdp:
70
+ fsdp_config:
71
+ resize_token_embeddings_to_32x: true
72
+ special_tokens:
73
+ bos_token: "<|endoftext|>"
74
+ eos_token: "<|endoftext|>"
75
+ unk_token: "<|endoftext|>"
76
+ pad_token: "<|pad|>"