Update README.md

README.md

@@ -13,8 +13,14 @@ tags:
 - writing
 - experimental
 - long-context
+datasets:
+- kalomaze/Opus_Instruct_25k
+- Fizzarolli/FallingThroughTheSkies-592k-Filtered-Filtered
+- Sao10K/c2-Logs-Filtered
+- nothingiisreal/Reddit-Dirty-And-WritingPrompts
 ---
 
+# Marlin v8: The Big Kahuna Update
 <img src="https://cdn-uploads.huggingface.co/production/uploads/662c17b252e194d5d436c708/ULeHz0KITPcS0znTN7gDl.png" width="500" height="500" />
 
 # Uploaded  model
@@ -23,26 +29,29 @@ tags:
 - **License:** apache-2.0
 - **Finetuned from model :** unsloth/Mistral-Nemo-Base-2407
 
-**Standard disclaimer:** This is me teaching myself the basics of fine-tuning, with notes extensively borrowed from https://huggingface.co/nothingiisreal/MN-12B-Celeste-V1.9
+**Standard disclaimer:** This is me teaching myself the basics of fine-tuning, with notes extensively borrowed from https://huggingface.co/nothingiisreal/MN-12B-Celeste-V1.9. Huge props to [nothingisreal](https://huggingface.co/nothingiisreal) for posting their process and making me think this was even possible for a little fish like me.
+
+The aim here is for a solid RP/storywriting model that will fit in 16GB of VRAM with a decent amount of context (> 16K).
 
 # New for v8:
 - Fine-tuned on Nemo Base instead of Instruct, because why not?
-- **FULL BORE MODE: ACTIVATE!** 10K-ish records of mostly-human convos and stories, curated by me, trained in ChatML, up from 8K in v6. Specifically:
+- **BIG KAHUNA POWERS: ACTIVATE!** 10K-ish records of mostly-human convos and stories, trained in ChatML, up from 8K in v6. For all of these records I did additional filtering/editing/selection beyond what I think happened in Celeste v1.9, mostly to teach myself some dataset skillz, plus I added more stories. Specifically:
   - 4K records from Reddit Writing Prompts (equal split of highest-rated sfw & nfsw)
-  - 2K of Claude instruct, lightly curated & de-clauded.
-  - 2K of curated Fallen Skies
+  - 2K of Claude instruct, lightly curated & de-clauded
+  - 2K of curated Falling through the Skies 
   - 2K of curated/lightly de-ministrated C2 chat
 - Trained on a single 80GB A100 from runpod.io, with batch size of 8 (up from 2 on A100 40G), so far less steps involved.
+- And remember kids, water is wet and fish are moist.
 
-I pulled v7 because I honestly don't think it's as good as v6, and don't want folks to get the wrong idea that it's better just because the version number is higher.
+I pulled v7 because I honestly don't think it's as good as v6, and don't want folks to get the wrong idea that it's better just because the version number is higher. Besides, nothing good ever fires on all _seven_ cylinders.
 
-Props again to Unsloth.ai for letting me train this on a single A100 with variable (wildly variable) context length. 
+Props again to [Daniel](https://huggingface.co/danielhanchen) and [Unsloth](https://huggingface.co/unsloth) for writing magic that lets me train this on a single A100 with variable (wildly variable) context length.
 
 Here's what the train/eval loss looked like:
 
-![image/png](https://cdn-uploads.huggingface.co/production/uploads/662c17b252e194d5d436c708/hUKuy7ht_qObuFNDTVEe9.png)
+<img src="https://cdn-uploads.huggingface.co/production/uploads/662c17b252e194d5d436c708/hUKuy7ht_qObuFNDTVEe9.png" width="800"/>
 
-I still don't know what makes training loss drop at the end of epoch 1, or why eval loss doesn't drop down to match (it continues to decrease, but slowly).
+I still don't know what makes training loss drop at the end of epoch 1, or why eval loss doesn't drop down to match (it continues to decrease, but slowly). I did say this was experimental, right?
 
 It was trained with the following settings:
 
@@ -91,4 +100,4 @@ lr_scheduler_kwargs = {
 
 This mistral model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
 
-[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)
+[<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)