ToastyPigeon commited on
Commit
706a4a5
·
verified ·
1 Parent(s): 1a62c60

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -174
README.md CHANGED
@@ -10,187 +10,24 @@ model-index:
10
  results: []
11
  ---
12
 
13
- <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
- should probably proofread and complete it, then remove this comment. -->
15
-
16
- [<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl)
17
- <details><summary>See axolotl config</summary>
18
-
19
- axolotl version: `0.4.1`
20
- ```yaml
21
- # Set up for use on 2x24gb cards
22
- # huggingface-cli login --token $hf_key && wandb login $wandb_key
23
- # python -m axolotl.cli.preprocess snm-adventure-s.yml
24
- # accelerate launch -m axolotl.cli.train snm-adventure-s.yml
25
- # python -m axolotl.cli.merge_lora snm-adventure-s.yml
26
- # huggingface-cli upload ToastyPigeon/snm-type1-adventure-s adventure-workspace/merged . --private
27
-
28
- base_model: arcee-ai/SuperNova-Medius
29
- model_type: AutoModelForCausalLM
30
- tokenizer_type: AutoTokenizer
31
-
32
- load_in_8bit: false
33
- load_in_4bit: true
34
- strict: false
35
- sequence_len: 8192
36
- min_sample_len: 128
37
- bf16: true
38
- fp16:
39
- tf32: false
40
- flash_attention: true
41
- special_tokens:
42
-
43
- # Data
44
- dataset_prepared_path: last_run_prepared
45
- datasets:
46
- - path: Columbidae/text-adventures
47
- type: completion
48
- warmup_steps: 20
49
- shuffle_merged_datasets: true
50
-
51
- save_safetensors: true
52
-
53
- # WandB
54
- wandb_project: SuperNova-Medius-SpringDragon
55
- wandb_entity:
56
-
57
- # Iterations
58
- num_epochs: 1
59
-
60
- # Output
61
- output_dir: ./adventure-workspace
62
- hub_model_id: ToastyPigeon/supernova-medius-adventure-s-qlora
63
- hub_strategy: "every_save"
64
-
65
- # Sampling
66
- sample_packing: true
67
- pad_to_sequence_len: true
68
-
69
- # Batching
70
- gradient_accumulation_steps: 2
71
- micro_batch_size: 1
72
- eval_batch_size: 1
73
- gradient_checkpointing: 'unsloth'
74
- gradient_checkpointing_kwargs:
75
- use_reentrant: true
76
-
77
- unsloth_cross_entropy_loss: true
78
- #unsloth_lora_mlp: true
79
- #unsloth_lora_qkv: true
80
- #unsloth_lora_o: true
81
-
82
- # Evaluation
83
- val_set_size: 100
84
- evals_per_epoch: 5
85
- eval_table_size:
86
- eval_max_new_tokens: 256
87
- eval_sample_packing: false
88
-
89
- # LoRA
90
- adapter: qlora
91
- lora_model_dir:
92
- lora_r: 64
93
- lora_alpha: 64
94
- lora_dropout: 0.125
95
- lora_target_linear:
96
- lora_fan_in_fan_out:
97
- lora_target_modules:
98
- - gate_proj
99
- - down_proj
100
- - up_proj
101
- - q_proj
102
- - v_proj
103
- - k_proj
104
- - o_proj
105
- lora_modules_to_save:
106
-
107
- # Optimizer
108
- optimizer: paged_adamw_8bit # adamw_8bit
109
- lr_scheduler: cosine
110
- learning_rate: 0.00005
111
- cosine_min_lr_ratio: 0.1
112
- weight_decay: 0.01
113
- max_grad_norm: 10.0
114
-
115
- # Misc
116
- train_on_inputs: false
117
- group_by_length: false
118
- early_stopping_patience:
119
- local_rank:
120
- logging_steps: 1
121
- xformers_attention:
122
- debug:
123
- deepspeed: /workspace/axolotl/deepspeed_configs/zero3.json # previously blank
124
- fsdp:
125
- fsdp_config:
126
-
127
- # Checkpoints
128
- resume_from_checkpoint:
129
- saves_per_epoch: 1
130
-
131
- plugins:
132
- - axolotl.integrations.liger.LigerPlugin
133
- liger_rope: true
134
- liger_rms_norm: true
135
- liger_swiglu: true
136
- liger_fused_linear_cross_entropy: true
137
- ```
138
-
139
- </details><br>
140
-
141
  # supernova-medius-adventure-s-qlora
142
 
143
- This model is a fine-tuned version of [arcee-ai/SuperNova-Medius](https://huggingface.co/arcee-ai/SuperNova-Medius) on the None dataset.
144
- It achieves the following results on the evaluation set:
145
- - Loss: 2.3335
146
-
147
- ## Model description
148
 
149
- More information needed
150
 
151
- ## Intended uses & limitations
152
 
153
- More information needed
154
 
155
- ## Training and evaluation data
156
 
157
- More information needed
158
-
159
- ## Training procedure
160
-
161
- ### Training hyperparameters
162
-
163
- The following hyperparameters were used during training:
164
- - learning_rate: 5e-05
165
- - train_batch_size: 1
166
- - eval_batch_size: 1
167
- - seed: 42
168
- - distributed_type: multi-GPU
169
- - num_devices: 2
170
- - gradient_accumulation_steps: 2
171
- - total_train_batch_size: 4
172
- - total_eval_batch_size: 2
173
- - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
174
- - lr_scheduler_type: cosine
175
- - lr_scheduler_warmup_steps: 20
176
- - num_epochs: 1
177
-
178
- ### Training results
179
-
180
- | Training Loss | Epoch | Step | Validation Loss |
181
- |:-------------:|:------:|:----:|:---------------:|
182
- | 2.4468 | 0.0052 | 1 | 2.5688 |
183
- | 2.3699 | 0.1995 | 38 | 2.3922 |
184
- | 2.3281 | 0.3990 | 76 | 2.3477 |
185
- | 2.2334 | 0.5984 | 114 | 2.3414 |
186
- | 2.3705 | 0.7979 | 152 | 2.3338 |
187
- | 2.2731 | 0.9974 | 190 | 2.3335 |
188
 
 
189
 
190
- ### Framework versions
 
191
 
192
- - PEFT 0.13.2
193
- - Transformers 4.45.2
194
- - Pytorch 2.4.1+cu124
195
- - Datasets 3.0.1
196
- - Tokenizers 0.20.1
 
10
  results: []
11
  ---
12
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  # supernova-medius-adventure-s-qlora
14
 
15
+ This model is a fine-tuned version of [arcee-ai/SuperNova-Medius](https://huggingface.co/arcee-ai/SuperNova-Medius) on old-style Adventure data.
 
 
 
 
16
 
17
+ I'm going to be honest with you - I don't remember if this is Spring Dragon or Skein+SpringDragon.
18
 
19
+ Based on the naming I assume it's Spring Dragon, but the axolotl config uploaded with it suggests it's both, so...
20
 
21
+ In either case, this is a QLoRA trained on SuperNova-Medius. It should work with other Qwen2.5 14B models as well.
22
 
23
+ User turns are prefaced with `>`, for example:
24
 
25
+ ```
26
+ The door in front of you is locked. You will need to find the key to proceed.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
27
 
28
+ > I search the desk in the corner
29
 
30
+ You search the drawers of the desk, but find nothing.
31
+ ```
32
 
33
+ I recommend setting `\n>` as a stopping string. This is the default behavior for KoboldAI Lite's Adventure mode.