SethGA commited on
Commit
1a9aeaf
1 Parent(s): 9f233bd

Model save

Browse files
Files changed (1) hide show
  1. README.md +138 -0
README.md ADDED
@@ -0,0 +1,138 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ tags:
4
+ - generated_from_trainer
5
+ base_model: NousResearch/Llama-2-7b-hf
6
+ model-index:
7
+ - name: neocortex-grounded
8
+ results: []
9
+ ---
10
+
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
+
14
+ [<img src="https://raw.githubusercontent.com/OpenAccess-AI-Collective/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/OpenAccess-AI-Collective/axolotl)
15
+ <details><summary>See axolotl config</summary>
16
+
17
+ axolotl version: `0.4.0`
18
+ ```yaml
19
+ base_model: NousResearch/Llama-2-7b-hf
20
+ model_type: LlamaForCausalLM
21
+ tokenizer_type: LlamaTokenizer
22
+ is_llama_derived_model: true
23
+ hub_model_id: neocortex-grounded
24
+
25
+ load_in_8bit: false
26
+ load_in_4bit: true
27
+ strict: false
28
+
29
+ datasets:
30
+ - path: SethGA/neocortex_grounded_23k
31
+ type: alpaca
32
+ shards: 20
33
+ dataset_prepared_path:
34
+ val_set_size: 0.05
35
+ output_dir: ./qlora-out
36
+
37
+ adapter: qlora
38
+ lora_model_dir:
39
+
40
+ sequence_len: 4096
41
+ sample_packing: false
42
+ eval_sample_packing: false
43
+ pad_to_sequence_len: true
44
+
45
+ lora_r: 32
46
+ lora_alpha: 16
47
+ lora_dropout: 0.05
48
+ lora_target_modules:
49
+ lora_target_linear: true
50
+ lora_fan_in_fan_out:
51
+
52
+ wandb_project: neocortex
53
+ wandb_entity:
54
+ wandb_watch:
55
+ wandb_run_id:
56
+ wandb_log_model: checkpoint
57
+
58
+ gradient_accumulation_steps: 4
59
+ micro_batch_size: 2
60
+ num_epochs: 3
61
+ optimizer: paged_adamw_32bit
62
+ lr_scheduler: cosine
63
+ learning_rate: 0.0002
64
+
65
+ train_on_inputs: false
66
+ group_by_length: false
67
+ bf16: auto
68
+ fp16:
69
+ tf32: false
70
+
71
+ gradient_checkpointing: true
72
+ early_stopping_patience:
73
+ resume_from_checkpoint:
74
+ local_rank:
75
+ logging_steps: 1
76
+ xformers_attention:
77
+ flash_attention: true
78
+
79
+ warmup_steps: 10
80
+ eval_steps: 20
81
+ eval_table_size: 5
82
+ save_strategy: epoch
83
+ save_steps:
84
+ debug:
85
+ deepspeed:
86
+ weight_decay: 0.0
87
+ fsdp:
88
+ fsdp_config:
89
+ special_tokens:
90
+ bos_token: "<s>"
91
+ eos_token: "</s>"
92
+ unk_token: "<unk>"
93
+ ```
94
+
95
+ </details><br>
96
+
97
+ # neocortex-grounded
98
+
99
+ This model is a fine-tuned version of [NousResearch/Llama-2-7b-hf](https://huggingface.co/NousResearch/Llama-2-7b-hf) on an unknown dataset.
100
+
101
+ ## Model description
102
+
103
+ More information needed
104
+
105
+ ## Intended uses & limitations
106
+
107
+ More information needed
108
+
109
+ ## Training and evaluation data
110
+
111
+ More information needed
112
+
113
+ ## Training procedure
114
+
115
+ ### Training hyperparameters
116
+
117
+ The following hyperparameters were used during training:
118
+ - learning_rate: 0.0002
119
+ - train_batch_size: 2
120
+ - eval_batch_size: 2
121
+ - seed: 42
122
+ - distributed_type: multi-GPU
123
+ - num_devices: 2
124
+ - gradient_accumulation_steps: 4
125
+ - total_train_batch_size: 16
126
+ - total_eval_batch_size: 4
127
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
128
+ - lr_scheduler_type: cosine
129
+ - lr_scheduler_warmup_steps: 10
130
+ - num_epochs: 3
131
+
132
+ ### Framework versions
133
+
134
+ - PEFT 0.9.0
135
+ - Transformers 4.39.0.dev0
136
+ - Pytorch 2.0.1+cu118
137
+ - Datasets 2.17.1
138
+ - Tokenizers 0.15.0