Update README.md
Browse files
README.md
CHANGED
@@ -14,6 +14,143 @@ library_name: transformers
|
|
14 |
This model was converted to GGUF format from [`PygmalionAI/Pygmalion-3-12B`](https://huggingface.co/PygmalionAI/Pygmalion-3-12B) using llama.cpp via the ggml.ai's [GGUF-my-repo](https://huggingface.co/spaces/ggml-org/gguf-my-repo) space.
|
15 |
Refer to the [original model card](https://huggingface.co/PygmalionAI/Pygmalion-3-12B) for more details on the model.
|
16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
17 |
## Use with llama.cpp
|
18 |
Install llama.cpp through brew (works on Mac and Linux)
|
19 |
|
|
|
14 |
This model was converted to GGUF format from [`PygmalionAI/Pygmalion-3-12B`](https://huggingface.co/PygmalionAI/Pygmalion-3-12B) using llama.cpp via the ggml.ai's [GGUF-my-repo](https://huggingface.co/spaces/ggml-org/gguf-my-repo) space.
|
15 |
Refer to the [original model card](https://huggingface.co/PygmalionAI/Pygmalion-3-12B) for more details on the model.
|
16 |
|
17 |
+
---
|
18 |
+
Model Details
|
19 |
+
-
|
20 |
+
|
21 |
+
|
22 |
+
|
23 |
+
It's been a long road fraught with delays, technical issues and us
|
24 |
+
banging our heads against the wall, but we're glad to say that we've
|
25 |
+
returned to open-source roleplaying with our newest model, Pygmalion-3.
|
26 |
+
We've taken Mistral's Nemo
|
27 |
+
base model and fed it hundreds of millions of tokens of conversations,
|
28 |
+
creative writing and instructions to create a model dedicated towards
|
29 |
+
roleplaying that we hope fulfills your expectations.
|
30 |
+
|
31 |
+
|
32 |
+
As part of our open-source roots and promises to those who have been
|
33 |
+
with us since the beginning, we release this model under the permissive Apache 2.0 license, allowing anyone to use and develop upon our work for everybody in the local models community.
|
34 |
+
|
35 |
+
|
36 |
+
|
37 |
+
|
38 |
+
|
39 |
+
|
40 |
+
|
41 |
+
Prompting
|
42 |
+
-
|
43 |
+
|
44 |
+
|
45 |
+
|
46 |
+
We've switched over to standard ChatML format for both convenience
|
47 |
+
and to allow easier merging with other ChatML-based models. Pygmalion-3
|
48 |
+
supports the same "Enter X mode" as our previous Pygmalion-2 model, but
|
49 |
+
experimentation with the system prompt is encouraged so that you can
|
50 |
+
find what is best for you.
|
51 |
+
|
52 |
+
|
53 |
+
Please note that some strange issues have been reported with
|
54 |
+
the <|im_end|> token. It is highly recommended to add a custom
|
55 |
+
token ban on the phrase "<|im_end|>" and "<" in general. We
|
56 |
+
apologize for the inconvenience.
|
57 |
+
|
58 |
+
|
59 |
+
|
60 |
+
|
61 |
+
|
62 |
+
|
63 |
+
|
64 |
+
Prompting example
|
65 |
+
-
|
66 |
+
|
67 |
+
|
68 |
+
|
69 |
+
<|im_start|>system
|
70 |
+
Enter roleplay mode. You shall reply to {{user}} while staying in character. Your responses must be detailed, creative, immersive, and drive the scenario forward. You will follow {{char}}'s persona.<|im_end|>
|
71 |
+
<|im_start|>user
|
72 |
+
{{user}}: Good evening!<|im_end|>
|
73 |
+
<|im_start|>assistant
|
74 |
+
{{char}}: It's three in the morning, man.<|im_end|>
|
75 |
+
|
76 |
+
|
77 |
+
|
78 |
+
Note that {{user}} and {{char}} are placeholders.
|
79 |
+
|
80 |
+
|
81 |
+
|
82 |
+
|
83 |
+
|
84 |
+
|
85 |
+
|
86 |
+
Dataset
|
87 |
+
-
|
88 |
+
|
89 |
+
|
90 |
+
|
91 |
+
We've gathered a large collection of instructions and roleplaying totaling hundreds of millions of tokens, including our PIPPA dataset and roleplaying forums.
|
92 |
+
|
93 |
+
|
94 |
+
|
95 |
+
|
96 |
+
|
97 |
+
|
98 |
+
|
99 |
+
Limitations and biases
|
100 |
+
-
|
101 |
+
|
102 |
+
|
103 |
+
|
104 |
+
The intended use-case for this model is fictional writing for entertainment purposes. Any other sort of usage is out of scope.
|
105 |
+
|
106 |
+
|
107 |
+
As such, it was not fine-tuned to be safe and
|
108 |
+
harmless: the base model and this fine-tune have been trained on data
|
109 |
+
known to contain profanity and texts that are lewd or otherwise
|
110 |
+
offensive. It may produce socially unacceptable or undesirable text,
|
111 |
+
even if the prompt itself does not include anything explicitly
|
112 |
+
offensive. Outputs might often be factually wrong or misleading.
|
113 |
+
|
114 |
+
|
115 |
+
|
116 |
+
|
117 |
+
|
118 |
+
|
119 |
+
|
120 |
+
Training Specifications
|
121 |
+
-
|
122 |
+
|
123 |
+
|
124 |
+
|
125 |
+
We trained our model as a rank-32 LoRA adapter with one epoch over
|
126 |
+
our data using 8x NVIDIA A40 GPUs. For this run, we employed a learning
|
127 |
+
rate of 2e-4 and a total batch size across all GPUs of 24. A cosine
|
128 |
+
learning rate scheduler was used with a 100 step warmup. DeepSpeed ZeRO
|
129 |
+
was used to successfully get memory usage down.
|
130 |
+
|
131 |
+
|
132 |
+
|
133 |
+
|
134 |
+
|
135 |
+
|
136 |
+
|
137 |
+
Acknowledgements
|
138 |
+
-
|
139 |
+
|
140 |
+
|
141 |
+
|
142 |
+
This project could not have been done without the compute support of Hive Digital Technologies and the Axolotl training software.
|
143 |
+
|
144 |
+
|
145 |
+
We'd like to extensively thank lemonilia for their wonderful help in compiling roleplay forum data.
|
146 |
+
|
147 |
+
|
148 |
+
And most of all, we dedicate this model to our great community,
|
149 |
+
who've stuck with us through everything until now. Sincerely, thank you
|
150 |
+
so much. We hope you enjoy our work to the fullest and we promise more
|
151 |
+
is on the way soon.
|
152 |
+
|
153 |
+
---
|
154 |
## Use with llama.cpp
|
155 |
Install llama.cpp through brew (works on Mac and Linux)
|
156 |
|