Upload ./README.md with huggingface_hub
Browse files
README.md
ADDED
@@ -0,0 +1,75 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
language:
|
4 |
+
- en
|
5 |
+
base_model:
|
6 |
+
- mistralai/Mistral-Nemo-Base-2407
|
7 |
+
tags:
|
8 |
+
- text adventure
|
9 |
+
- roleplay
|
10 |
+
library_name: transformers
|
11 |
+
---
|
12 |
+
### exl3 quant
|
13 |
+
---
|
14 |
+
### check revisions for quants
|
15 |
+
---
|
16 |
+
|
17 |
+
|
18 |
+

|
19 |
+
|
20 |
+
# Muse-12B
|
21 |
+
|
22 |
+
Muse brings an extra dimension to any tale—whether you're exploring a fantastical realm, court intrigue, or slice-of-life scenarios where a conversation can be as meaningful as a quest. While it handles adventure capably, Muse truly shines when character relationships and emotions are at the forefront, delivering impressive narrative coherence over long contexts.
|
23 |
+
|
24 |
+
If you want to easily try this model for free, you can do so at [https://aidungeon.com](https://aidungeon.com/).
|
25 |
+
|
26 |
+
We plan to continue improving and open-sourcing similar models, so please share any and all feedback on how we can improve model behavior. Below we share more details on how Muse was created.
|
27 |
+
|
28 |
+
[Quantized GGUF weights can be downloaded here.](https://huggingface.co/LatitudeGames/Muse-12B-GGUF)
|
29 |
+
|
30 |
+
## Model details
|
31 |
+
|
32 |
+
Muse 12B was trained using Mistral Nemo 12B as its foundation, with training occurring in three stages: SFT (supervised fine-tuning), followed by two distinct DPO (direct preference optimization) phases.
|
33 |
+
|
34 |
+
**SFT** - Various multi-turn datasets from a multitude of sources, combining text adventures of the kind used to finetune [our Wayfarer 12B model](https://huggingface.co/LatitudeGames/Wayfarer-12B), long emotional narratives and general roleplay, each carefully balanced and rewritten to be free of common AI cliches. A small single-turn instruct dataset was included to send a stronger signal during finetuning.
|
35 |
+
|
36 |
+
**DPO 1** - Gutenberg DPO, [credit to Jon Durbin](https://huggingface.co/datasets/jondurbin/gutenberg-dpo-v0.1) - This stage introduces human writing techniques, significantly enhancing the model's potential outputs, albeit trading some intelligence for the stylistic benefits of human-created text.
|
37 |
+
|
38 |
+
**DPO 2** - Reward Model User Preference Data, [detailed in our blog](https://blog.latitude.io/all-posts/synthetic-data-preference-optimization-and-reward-models) - This stage refines the Gutenberg stage's "wildness," restoring intelligence while maintaining enhanced writing quality and providing a final level of enhancement due to the reward model samples.
|
39 |
+
|
40 |
+
The result is a model that writes like no other: versatile across genres, natural in expression, and suited to emotional depth.
|
41 |
+
|
42 |
+
## Inference
|
43 |
+
|
44 |
+
The Nemo architecture is known for being sensitive to higher temperatures, so the following settings are recommended as a baseline. Nothing stops you from experimenting with these, of course.
|
45 |
+
|
46 |
+
```
|
47 |
+
"temperature": 0.8,
|
48 |
+
"repetition_penalty": 1.05,
|
49 |
+
"min_p": 0.025
|
50 |
+
```
|
51 |
+
|
52 |
+
## Limitations
|
53 |
+
|
54 |
+
Muse was trained exclusively on second-person present tense data (using “you”) in a narrative style. Other styles will work as well but may produce suboptimal results.
|
55 |
+
|
56 |
+
Average response lengths tend toward verbosity (1000+ tokens) due to the Gutenberg DPO influence, though this can be controlled through explicit instructions in the system prompt.
|
57 |
+
|
58 |
+
## Prompt Format
|
59 |
+
|
60 |
+
ChatML was used during all training stages.
|
61 |
+
|
62 |
+
```
|
63 |
+
<|im_start|>system
|
64 |
+
You're a masterful storyteller and gamemaster. Write in second person present tense (You are), crafting vivid, engaging narratives with authority and confidence.<|im_end|>
|
65 |
+
<|im_start|>user
|
66 |
+
> You peer into the darkness.
|
67 |
+
<|im_start|>assistant
|
68 |
+
You have been eaten by a grue.
|
69 |
+
|
70 |
+
GAME OVER
|
71 |
+
```
|
72 |
+
|
73 |
+
## Credits
|
74 |
+
|
75 |
+
Thanks to [Gryphe Padar](https://huggingface.co/Gryphe) for collaborating on this finetune with us!
|