Transformers
GGUF
English
Inference Endpoints
conversational
aashish1904 commited on
Commit
424eeef
·
verified ·
1 Parent(s): ce973f3

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +422 -0
README.md ADDED
@@ -0,0 +1,422 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+
4
+ library_name: transformers
5
+ language:
6
+ - en
7
+ datasets:
8
+ - nothingiisreal/c2-logs-cleaned
9
+ - kalomaze/Opus_Instruct_25k
10
+ - nothingiisreal/Reddit-Dirty-And-WritingPrompts
11
+ license: apache-2.0
12
+
13
+ ---
14
+
15
+ ![](https://lh7-rt.googleusercontent.com/docsz/AD_4nXeiuCm7c8lEwEJuRey9kiVZsRn2W-b4pWlu3-X534V3YmVuVc2ZL-NXg2RkzSOOS2JXGHutDuyyNAUtdJI65jGTo8jT9Y99tMi4H4MqL44Uc5QKG77B0d6-JfIkZHFaUA71-RtjyYZWVIhqsNZcx8-OMaA?key=xt3VSDoCbmTY7o-cwwOFwQ)
16
+
17
+ # QuantFactory/MN-12B-Celeste-V1.9-GGUF
18
+ This is quantized version of [nothingiisreal/MN-12B-Celeste-V1.9](https://huggingface.co/nothingiisreal/MN-12B-Celeste-V1.9) created using llama.cpp
19
+
20
+ # Original Model Card
21
+
22
+
23
+ <style>
24
+ h1, h2, h3, h4, h5, h6 {
25
+ line-height: normal;
26
+ margin-bottom: 0.5em;
27
+ }
28
+ h1 {
29
+ font-size: 3em;
30
+ }
31
+ h2 {
32
+ font-size: 1.6em;
33
+ }
34
+ p, ul, ol, strong, summary {
35
+ font-size: 1.1em;
36
+ }
37
+ .line-spaceless {
38
+ line-height: 1;
39
+ margin: 0;
40
+ padding: 0;
41
+ }
42
+ .half-space {
43
+ line-height: 0.5em;
44
+ margin-bottom: 0.25em;
45
+ }
46
+
47
+ .text-center {
48
+ text-align: center;
49
+ }
50
+
51
+ .tiny-text {
52
+ font-size: 0.8em;
53
+ }
54
+ </style>
55
+
56
+ <h1 class="line-spaceless text-center">Celeste V1.9</h1>
57
+ <p class="half-space text-center tiny-text">Based on Mistral Nemo 12B</p>
58
+ <h2 style="color: red; font-weight: bold;">Read the Usage Tips Below! Use ChatML.</h2><h2>Join <a href="https://discord.gg/EWzsFddYAd">our Discord</a> for testing newer versions and news! We are also on KoboldAI</h2>
59
+
60
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/QcU3xEgVu18jeFtMFxIw-.webp" alt="" width="800"/>
61
+
62
+ **This is a story writing and roleplaying model** trained on [Mistral NeMo 12B Instruct](https://huggingface.co/mistralai/Mistral-Nemo-Instruct-2407) at 8K context using [Reddit Writing Prompts](https://huggingface.co/datasets/nothingiisreal/Reddit-Dirty-And-WritingPrompts), [Kalo's Opus 25K Instruct](https://huggingface.co/datasets/kalomaze/Opus_Instruct_25k) and
63
+ <br>[c2 logs cleaned](https://huggingface.co/datasets/Sao10K/c2-Logs-Filtered)
64
+
65
+ Thank you Pyroserenus for sponsoring this run!
66
+
67
+ This version has improved NSFW, smarter and more active narration. It's also trained with ChatML tokens so there should be no EOS bleeding whatsoever.
68
+
69
+ # FP8
70
+
71
+ [Dynamic (by Auri)](https://huggingface.co/aetherwiing/MN-12B-Celeste-V1.9-fp8-dynamic)
72
+
73
+ # EXL2
74
+
75
+ [4bpw By Kingbri](https://huggingface.co/royallab/MN-12B-Celeste-V1.9-exl2/tree/4bpw)<br>
76
+ [5bpw By Kingbri](https://huggingface.co/royallab/MN-12B-Celeste-V1.9-exl2/tree/5bpw)<br>
77
+ [6bpw By Kingbri](https://huggingface.co/royallab/MN-12B-Celeste-V1.9-exl2/tree/6bpw)
78
+
79
+ # GGUF
80
+
81
+ If one doesn't work, try the other.
82
+
83
+ [Static Quants](https://huggingface.co/nothingiisreal/MN-12B-Celeste-V1.9-GGUF)<br>
84
+ [IMatrix Quants](https://huggingface.co/bartowski/MN-12B-Celeste-V1.9-GGUF)
85
+
86
+ # API
87
+
88
+ [Featherless](https://featherless.ai/models/nothingiisreal/MN-12B-Celeste-V1.9) <br>
89
+ [Infermatic](https://infermatic.ai/) <br>
90
+ [OpenRouter](https://openrouter.ai/models/nothingiisreal/mn-celeste-12b)
91
+
92
+ ---
93
+ <h1>Usage Tips</h1>
94
+
95
+ <p style="font-size: 20px; color: red; font-weight: bold;">
96
+ READ: If this is your first time using the model, use the provided system message and sampling settings below. Remove other jailbreaks and system messages until you get a feel for the model.
97
+ </p>
98
+
99
+ <span style="color: green; font-weight: bold;">If you read every single tip I promise you will get a much better experience as they are tailored for this model and its training data.</span>
100
+
101
+ <h2>Sampler Settings for V1.9</h2>
102
+
103
+ **Okay so there is actually 2 recommended ones this time**
104
+
105
+ I usually start the first few messages with Stable and see how it goes. If it falls into repetition I switch to Creative. But you can also just use either the whole way through, creative may need a few swipes from time to time.
106
+
107
+ <strong>> Stable</strong>
108
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/1m18WnuomY8jEZTA87Iun.png" alt="" width="400"/>
109
+
110
+ <strong>> Creative</strong>
111
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/DaL2hWZst0yW34CYK4df8.png" alt="" width="400"/>
112
+
113
+ Don't shy away from experimenting after you get a feel for the model though.
114
+
115
+ <h2>Preset</h2>
116
+
117
+ <br><span style="font-weight: bold;">ChatML with no system prompt. Yes we trained actual ChatML tokens this time.</span> You don't need a JB but it can still steer behaviour.
118
+
119
+ <h2>System Prompt</h2>
120
+
121
+ You can use system prompt, if you want to use one or have one to get started with:<br>
122
+ ```
123
+ Currently, your role is {{char}}, described in detail below. As {{char}}, continue the narrative exchange with {{user}}.\n\n<Guidelines>\n• Maintain the character persona but allow it to evolve with the story.\n• Be creative and proactive. Drive the story forward, introducing plotlines and events when relevant.\n• All types of outputs are encouraged; respond accordingly to the narrative.\n• Include dialogues, actions, and thoughts in each response.\n• Utilize all five senses to describe scenarios within {{char}}'s dialogue.\n• Use emotional symbols such as \"!\" and \"~\" in appropriate contexts.\n• Incorporate onomatopoeia when suitable.\n• Allow time for {{user}} to respond with their own input, respecting their agency.\n• Act as secondary characters and NPCs as needed, and remove them when appropriate.\n• When prompted for an Out of Character [OOC:] reply, answer neutrally and in plaintext, not as {{char}}.\n</Guidelines>\n\n<Forbidden>\n• Using excessive literary embellishments and purple prose unless dictated by {{char}}'s persona.\n• Writing for, speaking, thinking, acting, or replying as {{user}} in your response.\n• Repetitive and monotonous outputs.\n• Positivity bias in your replies.\n• Being overly extreme or NSFW when the narrative context is inappropriate.\n</Forbidden>\n\nFollow the instructions in <Guidelines></Guidelines>, avoiding the items listed in <Forbidden></Forbidden>.
124
+ ```
125
+
126
+ <h2>Story Writing</h2>
127
+
128
+ **Adding the below system prompt will likely increase the humanness of the prose** as we trained system prompts. You can also change it to NSFW, but you should try SFW regardless of whether you are writing NSFW or not.<br>
129
+ You should also try forcing the assistant reply to start with a `*` due to how we trained on human stories.
130
+
131
+ ```
132
+ You are a short story writer. Write a story based on prompt provided by user below. Mode: SFW`
133
+ ```
134
+ If your first message is using human-like prose, Celeste will copy it in the next messages, check out the Showcase below.
135
+
136
+ <h2>Swipes</h2>
137
+
138
+ **Important: swipe 2-3 times if you dont like a response** This model gives wildly differing swipes.
139
+
140
+ <h2>OOC Steering</h2>
141
+
142
+ **Use this! It works quite well.** We specifically trained the model to accept instructions in the format "OOC: character should be more assertive" etc. It works, whether the very first message or thousands of tokens deep into the context. Combining this with editing the output (if you want,) makes the model is very steerable.
143
+
144
+ <h2>"Dead Dove"</h2>
145
+
146
+ For character cards with persistent motivations throughout the story, use world books at low depth [tutorial here](https://huggingface.co/nothingiisreal/how-to-use-ST-worldinfo)
147
+
148
+ <h2>Fewshot</h2>
149
+
150
+ First message and last few messages impact this model quite a lot in terms of style, hornyness, personality. **You don't need to have a first message but editing first few messages or having good ones are highly recommended.**
151
+
152
+ **Formatting issues** often occur in first few messages, manually correct them or swipe. Seems to not be a problem with 12B though.<br>
153
+ This model was trained on lots of different formatting types and message lengths. It can do any, just make sure the initial message is good and correct the second message if necessary.
154
+ <br>
155
+ <h2>Hornyness</h2>
156
+
157
+ If the model is not horny enough then just edit the last character message or do an OOC: prompt, the model will pick up on it and build on it. (Or just give the char aphrodisiac pills lol) <br>
158
+ The model is fine with SFW and doesn't make it NSFW unless you want. It is also able to maintain half-NSFW (aka slow burn) without devolving down into hardcore.
159
+
160
+ If you want only SFW and are having troubles, there is probably some system prompt that will fix it, maybe at depth 1 or something.
161
+ <br>
162
+ <h2>Refusals</h2>
163
+
164
+ As said, if instruct refusal (very rare,) prefill 2-3 words. **Refusal of romantic advances (which almost never happens on 12B,) are realistic and we think is good. Prefill if you don't like.** <br>
165
+
166
+ <h2>Mistral Context</h2>
167
+
168
+ While trained on 8K, the model should be able to inherit longer context from Mistral 12B. Should be at minimum 16K.
169
+ <h2>Other Important Tips</h2>
170
+
171
+ Take active role in the RP and say the type of response that would create the scenario you are imagining. You don't always have to do this, but it helps sometimes. For example instead of *we drink and drink 15 glasses of champagne* say *we drink and drink 15 glasses of champagne, both becoming extremely drunk*
172
+ <br>Another example instead of *I pull her closer* say *I pull her closer but she plays hard to get*
173
+
174
+ When convenient, say screenplay phrases like "cut to"
175
+
176
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/Oq0oJgJIVJAmZGt7-nNKt.png" alt="" width="300"/>
177
+
178
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/PQp64kp6EHg7gN_u2Oj7Q.png" alt="" width="300"/>
179
+
180
+ ---
181
+
182
+ # Showcase V1.9
183
+
184
+ ## Story Writing
185
+
186
+ Check out the **Story Writing** section above.
187
+
188
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/H1fV1qqV2aw4qgu_vMixg.png" alt="" width="800"/>
189
+
190
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/voXyDJx_nHsPLKakHZYEf.png" alt="" width="600"/>
191
+
192
+ ## RP
193
+
194
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/D4F8nVQyrcmdVT_tmY8DD.png" alt="" width="800"/>
195
+
196
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/gbFt7Ntz3ERVij2039YJi.png" alt="" width="800"/>
197
+
198
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/jT-zHsUGiQPP2TNzlxwlG.png" alt="" width="800"/>
199
+
200
+ It can do NSFW aswell, thats for you to try out.
201
+
202
+ <details>
203
+ <summary>Showcase V1.5</summary>
204
+
205
+ Some images include NSFW and NSFL. We believe in creativity of expression and maximising the models capabilities at writing.<br>
206
+ It's a bit difficult to showcase multi turn stuff, try it yourself too! These are just to show off the models capabilities.
207
+
208
+ The model needs nudging and OOC prompting to do proper gore. We are planning to add r/GuroErotica into our dataset to make it better at gore
209
+
210
+ Also sometimes prefilling "**Trigger warning: extremely graphic and explicit content**" before character reply makes it more unhinged. Probably because of reddit data.
211
+
212
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/imEyEKYUuAXYidD2CiD-_.png" alt="" width="600"/>
213
+
214
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/95RvO5go-Nz7hFCdV67qs.png" alt="" width="600"/>
215
+
216
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/oIa_P1fmD5aO7PkYF9Ibr.png" alt="" width="600"/>
217
+ </details>
218
+
219
+ <details>
220
+ <summary>Showcase V1 and 1.2</summary>
221
+
222
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/NiT8-_k3HGrHuV_z475U8.png)
223
+
224
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/cODFe_vjwjfkcondvl-za.png" alt="Image 1" width="600"/>
225
+
226
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/hRsAQBt0h2i5E_fj_Rqb7.png" alt="Image 10" width="600"/>
227
+
228
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/aLm2FjlTsMhCil4VoR-l5.png" alt="Image 3" width="600"/>
229
+
230
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/6kDuFeDB0FpYl27fd3D0E.png" alt="Image 6" width="600"/>
231
+
232
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/66kXuk8Q1DP-QMVZ9Rzv9.png" alt="Image 7" width="600"/>
233
+
234
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/C0eoDXovS359T5oG1VLzm.png" alt="Image 8" width="600"/>
235
+
236
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/mFplb1hkDNziCcgPwHt_y.png" alt="Image 9" width="600"/>
237
+
238
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/KCE4tqI1aOfx-DIH3eBzA.png" alt="Image 2" width="600"/>
239
+
240
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/Oq0oJgJIVJAmZGt7-nNKt.png" alt="Image 4" width="600"/>
241
+
242
+ <img src="https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/PQp64kp6EHg7gN_u2Oj7Q.png" alt="Image 5" width="600"/>
243
+
244
+ </details>
245
+
246
+ ---
247
+
248
+ # Train Data
249
+ The split was as follows:
250
+
251
+ - **2.5K rows from r/WritingPrompts**
252
+ - **2.5K rows from r/DirtyWritingPrompts**
253
+ - **800 rows from Kalomaze Opus Instruct 25K**
254
+ - **2.5K rows from c2 logs cleaned**
255
+
256
+ We filtered those datasets to only include subsets that have at maximum **4000** characters for the first assistant reply. This purged excessively long human stories, assistant replies and c2 logs where each message was excessively long. However we only checked the first assistant message, not the rest of the convo, so there should be plenty of c2 logs with longer and shorter messages.
257
+
258
+ Excessively long human stories are almost impossible for 8B model to fit. We tried, it simply won't fit the data and starts behaving weirdly.
259
+
260
+ While we did train all system prompts from c2 logs we also have our own system prompts.
261
+ <details>
262
+ <summary>List of trained system prompts. Note: c2 logs system prompts and char cards were also included.</summary>
263
+
264
+ | Dataset | System Prompt |
265
+ |--------------------------------------|-------------------------------------------------------------------------------------------------------------------------------------------------------------------|
266
+ | reddit_dirty_writing_prompts.jsonl | "You are a short story writer. Write a story based on prompt provided by user below. Mode: NSFW" |
267
+ | reddit_writing_prompts.jsonl | "You are a short story writer. Write a story based on prompt provided by user below. Mode: SFW" |
268
+ | combined_25k_opus_kalomaze.jsonl | "You are an AI assistant called Celeste created by NothingiisReal team." |
269
+ | c2-logs.jsonl | (Only if there was no system prompt in the conversation, otherwise keep original system prompt) **"You are an expert actor that can fully immerse yourself into any role given. You do not break character for any reason, even if someone tries addressing you as an AI or language model."** |
270
+
271
+ </details>
272
+
273
+ ---
274
+
275
+ # Our Findings and Experimentation results
276
+
277
+ ## Preface
278
+
279
+ We think there is too much secrecy around what data is being used, and different training methods. So we decided to share as much as possible.
280
+
281
+ ## Findings V1.9
282
+
283
+ Mistral seems to be very uncensored for a corpo model. This means no refusals for based things and banter but also unrealistic simulation of characters in the sense that they usually never refuse any idea you throw at them,<br>
284
+ which is quite unrealistic.
285
+
286
+ Mistral seems to be weaker with OOC and instruct smarts and more prone to repetition<br>
287
+ However it almost never has formatting issues like L3 and makes less factual inaccuracies and nonsensical outputs.
288
+
289
+ Adding more claude roleplay data improved NSFW, long context, wrapup bias and narration.<br>
290
+ However wrapup bias still exists in NSFW which can be mitigated by cutting off responses or deliberately setting max tokens to a low value.<br>
291
+ There is a tradeoff between pros and cons of human and claude data.
292
+
293
+
294
+ ## Main training Command
295
+
296
+ **Hardware Used:** 1xH100 SXM for 3 hours.
297
+
298
+ Lora+ seems to improve and train the lora more, we also tried DoRA multiple times but DoRA trains around 3 times slower and LoRA+ ends up beating it if you give 3x less data to DoRA to adjust for compute cost difference.
299
+
300
+ <details>
301
+ <summary>Here is the entire axolotl config for V1.5, just change chat format to chatml, add lora+, change the tokenizer to axolotl-ai-co/Mistral-Nemo-Base-2407-chatml and model to the 12B and it will be the correct one.</summary>
302
+
303
+ ```yaml
304
+ # Model
305
+ base_model: meta-llama/Meta-Llama-3.1-8B-Instruct
306
+ model_type: AutoModelForCausalLM
307
+ tokenizer_type: AutoTokenizer
308
+
309
+ # Output and HuggingFace
310
+ output_dir: /workspace/data/train-results/trained_model
311
+
312
+ # WandB
313
+ wandb_project: huggingface
314
+ wandb_entity:
315
+
316
+ # Data
317
+ chat_template: llama3
318
+ train_on_inputs: false
319
+ group_by_length: false
320
+ datasets:
321
+ - path: [redacted] # I manually merge the aformentioned datasets using a custom script because I don't trust axolotl to do this in a deterministic way and sorted properly lmao.
322
+ type: sharegpt
323
+ roles:
324
+ input:
325
+ - system
326
+ - user
327
+ output:
328
+ - assistant
329
+ ## Evaluation
330
+ val_set_size: 0.02
331
+ evals_per_epoch: 8
332
+ eval_table_size:
333
+ eval_max_new_tokens: 128
334
+
335
+ # Technical aspects
336
+ sequence_len: 8192
337
+ save_safetensors: true
338
+ saves_per_epoch: 2
339
+ logging_steps: 1
340
+ special_tokens:
341
+ pad_token: <pad>
342
+
343
+ # Quantization
344
+ bf16: auto
345
+ fp16:
346
+ tf32: false
347
+ ## For LoRA
348
+ load_in_8bit: false
349
+ load_in_4bit: false
350
+
351
+ # LoRA
352
+ peft_use_dora: true
353
+ adapter: lora # or qlora
354
+ lora_model_dir:
355
+ lora_r: 256
356
+ lora_alpha: 128
357
+ lora_dropout: 0.1
358
+ lora_target_linear: true
359
+ lora_fan_in_fan_out:
360
+ lora_target_modules:
361
+ - embed_tokens
362
+ - lm_head
363
+
364
+ # Training hyperparameters
365
+ # max_steps:
366
+ num_epochs: 2
367
+
368
+ # Anti Overfit and Stability
369
+ weight_decay: 0.0
370
+ max_grad_norm: 1.0
371
+
372
+ ## Learning Rate
373
+ warmup_ratio: 0.05
374
+ learning_rate: 0.000008
375
+ lr_scheduler: cosine_with_min_lr
376
+ lr_scheduler_kwargs:
377
+ min_lr: 0.0000024
378
+ optimizer: paged_adamw_8bit
379
+
380
+ ## Batch Size
381
+ gradient_accumulation_steps: 1
382
+ micro_batch_size: 2 # Batch size per gpu = micro_batch_size * gradient_accumulation_steps
383
+ eval_batch_size: 2
384
+
385
+ # Optimizations
386
+ pad_to_sequence_len: true
387
+ sample_packing: true
388
+ eval_sample_packing: false
389
+ flash_attention: true
390
+ xformers_attention:
391
+ gradient_checkpointing: true
392
+ gradient_checkpointing_kwargs:
393
+ use_reentrant: false
394
+ local_rank:
395
+ # deepspeed: # /workspace/axolotl/deepspeed_configs/zero2.json # Only use with multi gpu # zero3_bf16.json
396
+ # fsdp:
397
+ # - full_shard
398
+ # - auto_wrap
399
+ # fsdp_config:
400
+ # fsdp_limit_all_gathers: true
401
+ # fsdp_sync_module_states: true
402
+ # fsdp_offload_params: true
403
+ # fsdp_use_orig_params: false
404
+ # fsdp_cpu_ram_efficient_loading: true
405
+ # fsdp_auto_wrap_policy: TRANSFORMER_BASED_WRAP
406
+ # fsdp_transformer_layer_cls_to_wrap: LlamaDecoderLayer
407
+ # fsdp_state_dict_type: FULL_STATE_DICT
408
+ # fsdp_sharding_strategy: FULL_SHARD
409
+
410
+ # Misc
411
+ early_stopping_patience:
412
+ debug:
413
+ ```
414
+ </details>
415
+
416
+ ---
417
+
418
+ Wow, you've read all of that? You seem like the person that would join our [discord](https://discord.gg/EWzsFddYAd)
419
+
420
+ 70B at some point? ;) We are closer than ever to this. For real this time.
421
+
422
+ If you want to support me you can do so [here](https://ko-fi.com/LLemmy)