Text Generation
Transformers
Safetensors
English
llama
llama-factory
Not-For-All-Audiences
conversational
text-generation-inference
Inference Endpoints
aaronday3 commited on
Commit
ddccfca
1 Parent(s): 600f42a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +132 -171
README.md CHANGED
@@ -2,199 +2,160 @@
2
  library_name: transformers
3
  tags:
4
  - llama-factory
 
 
 
 
5
  ---
6
 
7
- # Model Card for Model ID
8
 
9
- <!-- Provide a quick summary of what the model is/does. -->
10
 
11
 
 
12
 
13
- ## Model Details
14
 
15
- ### Model Description
 
 
 
 
16
 
17
- <!-- Provide a longer summary of what this model is. -->
18
 
19
- This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
20
 
21
- - **Developed by:** [More Information Needed]
22
- - **Funded by [optional]:** [More Information Needed]
23
- - **Shared by [optional]:** [More Information Needed]
24
- - **Model type:** [More Information Needed]
25
- - **Language(s) (NLP):** [More Information Needed]
26
- - **License:** [More Information Needed]
27
- - **Finetuned from model [optional]:** [More Information Needed]
28
 
29
- ### Model Sources [optional]
30
 
31
- <!-- Provide the basic links for the model. -->
32
-
33
- - **Repository:** [More Information Needed]
34
- - **Paper [optional]:** [More Information Needed]
35
- - **Demo [optional]:** [More Information Needed]
36
-
37
- ## Uses
38
-
39
- <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
40
-
41
- ### Direct Use
42
-
43
- <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
44
-
45
- [More Information Needed]
46
-
47
- ### Downstream Use [optional]
48
-
49
- <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
50
-
51
- [More Information Needed]
52
-
53
- ### Out-of-Scope Use
54
-
55
- <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
56
-
57
- [More Information Needed]
58
-
59
- ## Bias, Risks, and Limitations
60
-
61
- <!-- This section is meant to convey both technical and sociotechnical limitations. -->
62
-
63
- [More Information Needed]
64
-
65
- ### Recommendations
66
-
67
- <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
68
-
69
- Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
70
-
71
- ## How to Get Started with the Model
72
-
73
- Use the code below to get started with the model.
74
-
75
- [More Information Needed]
76
-
77
- ## Training Details
78
-
79
- ### Training Data
80
-
81
- <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
82
-
83
- [More Information Needed]
84
-
85
- ### Training Procedure
86
-
87
- <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
88
-
89
- #### Preprocessing [optional]
90
-
91
- [More Information Needed]
92
-
93
-
94
- #### Training Hyperparameters
95
-
96
- - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
97
-
98
- #### Speeds, Sizes, Times [optional]
99
-
100
- <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
101
-
102
- [More Information Needed]
103
-
104
- ## Evaluation
105
-
106
- <!-- This section describes the evaluation protocols and provides the results. -->
107
-
108
- ### Testing Data, Factors & Metrics
109
-
110
- #### Testing Data
111
-
112
- <!-- This should link to a Dataset Card if possible. -->
113
-
114
- [More Information Needed]
115
-
116
- #### Factors
117
-
118
- <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
119
-
120
- [More Information Needed]
121
-
122
- #### Metrics
123
-
124
- <!-- These are the evaluation metrics being used, ideally with a description of why. -->
125
-
126
- [More Information Needed]
127
-
128
- ### Results
129
-
130
- [More Information Needed]
131
-
132
- #### Summary
133
-
134
-
135
-
136
- ## Model Examination [optional]
137
-
138
- <!-- Relevant interpretability work for the model goes here -->
139
-
140
- [More Information Needed]
141
-
142
- ## Environmental Impact
143
-
144
- <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
145
-
146
- Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
147
-
148
- - **Hardware Type:** [More Information Needed]
149
- - **Hours used:** [More Information Needed]
150
- - **Cloud Provider:** [More Information Needed]
151
- - **Compute Region:** [More Information Needed]
152
- - **Carbon Emitted:** [More Information Needed]
153
-
154
- ## Technical Specifications [optional]
155
-
156
- ### Model Architecture and Objective
157
-
158
- [More Information Needed]
159
-
160
- ### Compute Infrastructure
161
-
162
- [More Information Needed]
163
-
164
- #### Hardware
165
-
166
- [More Information Needed]
167
-
168
- #### Software
169
-
170
- [More Information Needed]
171
-
172
- ## Citation [optional]
173
-
174
- <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
175
 
176
- **BibTeX:**
 
177
 
178
- [More Information Needed]
 
 
 
179
 
180
- **APA:**
 
 
181
 
182
- [More Information Needed]
 
 
 
183
 
184
- ## Glossary [optional]
 
 
 
185
 
186
- <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
 
 
187
 
188
- [More Information Needed]
 
 
 
189
 
190
- ## More Information [optional]
191
 
192
- [More Information Needed]
193
 
194
- ## Model Card Authors [optional]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
195
 
196
- [More Information Needed]
197
 
198
- ## Model Card Contact
199
 
200
- [More Information Needed]
 
2
  library_name: transformers
3
  tags:
4
  - llama-factory
5
+ - not-for-all-audiences
6
+ license: llama3
7
+ language:
8
+ - en
9
  ---
10
 
11
+ # L3 8B Celeste
12
 
 
13
 
14
 
15
+ We trained [LLaMA 3 8B Instruct](https://huggingface.co/meta-llama/Meta-Llama-3-8B-Instruct) at 8K context using [Reddit Writing Prompts](https://huggingface.co/datasets/nothingiisreal/Reddit-Dirty-And-WritingPrompts), [Opus 15K Instruct](https://huggingface.co/datasets/nothingiisreal/Claude-3-Opus-Instruct-15K) and [c2 logs cleaned](https://huggingface.co/datasets/nothingiisreal/c2-logs-cleaned)
16
 
17
+ # Usage Tips
18
 
19
+ ## System Message / Jailbreak
20
+ <p style="font-size: 20px; color: red; font-weight: bold;">
21
+ IF THE GENERATIONS ARE BAD, REMOVE ALL SYSTEM PROMPTS
22
+ </p>
23
+ Particularly SillyTavern default prompts can make the model worse. Claude Jailbreaks should work fine though, there was a lot of them in c2 logs.
24
 
25
+ **You don't need a JB for casual usage but a JB can steer behaviour still.**
26
 
27
+ ## Sampling
28
 
29
+ You **should** also try messing with different settings.
 
 
 
 
 
 
30
 
31
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/uzVgp1ZMNV_LRx1stLxJ6.png)
32
 
33
+ ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
34
 
35
+ # Train Data
36
+ The split was as follows:
37
 
38
+ - **2K rows from r/WritingPrompts**
39
+ - **2K rows from r/DirtyWritingPrompts**
40
+ - **2K rows from Opus Instruct 15K (specifically the 6.5K jsonl)**
41
+ - **2K rows from c2 logs cleaned**
42
 
43
+ While we did train all system prompts from c2 logs we also have our own system prompts.
44
+ <details>
45
+ <summary>List of trained system prompts. Note: c2 logs system prompts and char cards were also included.</summary>
46
 
47
+ ### reddit_dirty_writing_prompts.jsonl
48
+ **2000**
49
+ **"You are a short story writer. Write a story based on prompt provided by user below.
50
+ Mode: NSFW"**
51
 
52
+ ### reddit_writing_prompts.jsonl
53
+ **2000**
54
+ **"You are a short story writer. Write a story based on prompt provided by user below.
55
+ Mode: SFW"**
56
 
57
+ ### Opus_Instruct-v2-6.5K-Filtered-v2.jsonl
58
+ **2000**
59
+ **""** (no prompt)
60
 
61
+ ### deduped-c2-logs-maywell-final-filter-4.jsonl
62
+ **2000**
63
+ (Only if there was no system prompt in the conversation, otherwise keep original system prompt)
64
+ **"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."**
65
 
66
+ </details>
67
 
68
+ ---
69
 
70
+ # Our Findings and Experimentation results
71
+
72
+ ## Preface
73
+
74
+ 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.
75
+
76
+ ## Findings
77
+
78
+ ### The Good
79
+ We found that increasing the amount of ranks from 64 to 256 has reduced repetition but also led to the language used resembling Claude more than the 64 rank version. No worries, it's still far enough from Claude.
80
+
81
+ It also led to increased coherency but reduced instruction following, likely because the model started diverging more away from L3 8B Instruct.
82
+
83
+ **The model is uncensored for RP. For Instruct it needs 2-3 words of prefill for the first message.**
84
+
85
+ We found that increasing the amount of data from 1K to 8K reduced repetition aswell.
86
+
87
+ **The prose is much better than other synthetic data generations. The model also demonstrates increased style copying abilities likely a result of the long data and varying writing styles found in WritingPrompts.**
88
+
89
+ **The model is exceptional at being creative in roleplaying**, knows different persona's and even a single character will change persona in different contexts, persona is tied to last few messages rather than system message or character card. **This is great as it often means the model can do impressive things without you needing to explicitly specify.**
90
+
91
+ ### Improvements for Next Run
92
+
93
+ Formatting can break sometimes.
94
+
95
+ ### Comments about training
96
+
97
+ Grad norm kept increasing throughout the run which is concerning, albeit it could be a side effect of the LR getting lower due to cosine LR Scheduler.
98
+
99
+ ## Graphs
100
+ Colors:
101
+ <p style="color: #F0B899;">256 rank on 8K rows</p>
102
+ <p style="color: #5BC5DB;">64 rank on 8K rows</p>
103
+ <p style="color: #5387DD;">64 rank on 1K rows</p>
104
+
105
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/630cf5d14ca0a22768bbe10c/y9hC4bGq-Lt7sDQ23q5db.png)
106
+
107
+ ## Main training Command
108
+
109
+ **Hardware Used:** 4xH100 NVL for 2 hours.
110
+
111
+ Here is the command, edit rank, learning rate, and any other parameter as you wish.
112
+ ```
113
+ !FORCE_TORCHRUN=1 llamafactory-cli train \
114
+ --stage sft \
115
+ --do_train True \
116
+ --model_name_or_path NousResearch/Meta-Llama-3-8B-Instruct \
117
+ --preprocessing_num_workers 16 \
118
+ --finetuning_type lora \
119
+ --quantization_method bitsandbytes \
120
+ --use_rslora False \
121
+ --lora_rank 64 \
122
+ --lora_alpha 64 \
123
+ --lora_dropout 0.1 \
124
+ --lora_target all \
125
+ --template llama3 \
126
+ --flash_attn fa2 \
127
+ --deepspeed examples/deepspeed/ds_z3_config.json \
128
+ --use_unsloth False \
129
+ --dataset_dir /workspace/sft \
130
+ --dataset dataset_name \
131
+ --cutoff_len 8192 \
132
+ --learning_rate 4e-6 \
133
+ --lr_scheduler_type cosine \
134
+ --num_train_epochs 2.0 \
135
+ --max_samples 100000 \
136
+ --per_device_train_batch_size 2 \
137
+ --gradient_accumulation_steps 1 \
138
+ --logging_steps 3 \
139
+ --save_steps 500 \
140
+ --warmup_ratio 0.05 \
141
+ --val_size 50 \
142
+ --eval_strategy steps \
143
+ --eval_steps 0.05 \
144
+ --optim adamw_bnb_8bit \
145
+ --packing False \
146
+ --train_on_prompt False \
147
+ --report_to all \
148
+ --max_grad_norm 1.0 \
149
+ --output_dir saves/LLaMA3-8B/trained-models/8krows-dwrp-c2l-opus-lora-4e-6-cosine-24-normal-bs \
150
+ --bf16 True \
151
+ --plot_loss True \
152
+ --ddp_timeout 180000000 \
153
+ --per_device_eval_batch_size 4 \
154
+ --include_num_input_tokens_seen True
155
+ ```
156
 
157
+ ---
158
 
159
+ Wow, you've read all of that? You seem like the person that would join our [discord](https://discord.gg/YcrXhk7QD7)
160
 
161
+ Gemma 9B and 27B at some point? ;)