Text Generation
Transformers
Safetensors
English
mistral
text2text-generation
conversational
text-generation-inference
rainy69 Juyoung Suk commited on
Commit
63f9c9b
·
verified ·
0 Parent(s):

Duplicate from prometheus-eval/prometheus-7b-v2.0

Browse files

Co-authored-by: Juyoung Suk <scottsuk0306@users.noreply.huggingface.co>

.gitattributes ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,182 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - text2text-generation
4
+ datasets:
5
+ - prometheus-eval/Feedback-Collection
6
+ - prometheus-eval/Preference-Collection
7
+ license: apache-2.0
8
+ language:
9
+ - en
10
+ pipeline_tag: text2text-generation
11
+ library_name: transformers
12
+ metrics:
13
+ - pearsonr
14
+ - spearmanr
15
+ - kendall-tau
16
+ - accuracy
17
+ ---
18
+ ## Links for Reference
19
+
20
+ - **Homepage: In Progress**
21
+ - **Repository:https://github.com/prometheus-eval/prometheus-eval**
22
+ - **Paper:https://arxiv.org/abs/2405.01535**
23
+ - **Point of Contact:seungone@cmu.edu**
24
+
25
+ # TL;DR
26
+ Prometheus 2 is an alternative of GPT-4 evaluation when doing fine-grained evaluation of an underlying LLM & a Reward model for Reinforcement Learning from Human Feedback (RLHF).
27
+ ![plot](./finegrained_eval.JPG)
28
+
29
+ Prometheus 2 is a language model using [Mistral-Instruct](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2) as a base model.
30
+ It is fine-tuned on 100K feedback within the [Feedback Collection](https://huggingface.co/datasets/prometheus-eval/Feedback-Collection) and 200K feedback within the [Preference Collection](https://huggingface.co/datasets/prometheus-eval/Preference-Collection).
31
+ It is also made by weight merging to support both absolute grading (direct assessment) and relative grading (pairwise ranking).
32
+ The surprising thing is that we find weight merging also improves performance on each format.
33
+
34
+ # Model Details
35
+
36
+ ## Model Description
37
+
38
+ - **Model type:** Language model
39
+ - **Language(s) (NLP):** English
40
+ - **License:** Apache 2.0
41
+ - **Related Models:** [All Prometheus Checkpoints](https://huggingface.co/models?search=prometheus-eval/Prometheus)
42
+ - **Resources for more information:**
43
+ - [Research paper](https://arxiv.org/abs/2405.01535)
44
+ - [GitHub Repo](https://github.com/prometheus-eval/prometheus-eval)
45
+
46
+
47
+ Prometheus is trained with two different sizes (7B and 8x7B).
48
+ You could check the 8x7B sized LM on [this page](https://huggingface.co/prometheus-eval/prometheus-2-8x7b-v2.0).
49
+ Also, check out our dataset as well on [this page](https://huggingface.co/datasets/prometheus-eval/Feedback-Collection) and [this page](https://huggingface.co/datasets/prometheus-eval/Preference-Collection).
50
+
51
+ ## Prompt Format
52
+
53
+ We have made wrapper functions and classes to conveniently use Prometheus 2 at [our github repository](https://github.com/prometheus-eval/prometheus-eval).
54
+ We highly recommend you use it!
55
+
56
+ However, if you just want to use the model for your use case, please refer to the prompt format below.
57
+ Note that absolute grading and relative grading requires different prompt templates and system prompts.
58
+
59
+ ### Absolute Grading (Direct Assessment)
60
+ Prometheus requires 4 components in the input: An instruction, a response to evaluate, a score rubric, and a reference answer. You could refer to the prompt format below.
61
+ You should fill in the instruction, response, reference answer, criteria description, and score description for score in range of 1 to 5.
62
+
63
+ Fix the components with \{text\} inside.
64
+ ```
65
+ ###Task Description:
66
+ An instruction (might include an Input inside it), a response to evaluate, a reference answer that gets a score of 5, and a score rubric representing a evaluation criteria are given.
67
+ 1. Write a detailed feedback that assess the quality of the response strictly based on the given score rubric, not evaluating in general.
68
+ 2. After writing a feedback, write a score that is an integer between 1 and 5. You should refer to the score rubric.
69
+ 3. The output format should look as follows: \"Feedback: (write a feedback for criteria) [RESULT] (an integer number between 1 and 5)\"
70
+ 4. Please do not generate any other opening, closing, and explanations.
71
+
72
+ ###The instruction to evaluate:
73
+ {orig_instruction}
74
+
75
+ ###Response to evaluate:
76
+ {orig_response}
77
+
78
+ ###Reference Answer (Score 5):
79
+ {orig_reference_answer}
80
+
81
+ ###Score Rubrics:
82
+ [{orig_criteria}]
83
+ Score 1: {orig_score1_description}
84
+ Score 2: {orig_score2_description}
85
+ Score 3: {orig_score3_description}
86
+ Score 4: {orig_score4_description}
87
+ Score 5: {orig_score5_description}
88
+
89
+ ###Feedback:
90
+ ```
91
+
92
+ After this, you should apply the conversation template of Mistral (not applying it might lead to unexpected behaviors).
93
+ You can find the conversation class at this [link](https://github.com/lm-sys/FastChat/blob/main/fastchat/conversation.py).
94
+ ```
95
+ conv = get_conv_template("mistral")
96
+ conv.set_system_message("You are a fair judge assistant tasked with providing clear, objective feedback based on specific criteria, ensuring each assessment reflects the absolute standards set for performance.")
97
+ conv.append_message(conv.roles[0], dialogs['instruction'])
98
+ conv.append_message(conv.roles[1], None)
99
+ prompt = conv.get_prompt()
100
+
101
+ x = tokenizer(prompt,truncation=False)
102
+ ```
103
+
104
+ As a result, a feedback and score decision will be generated, divided by a separating phrase ```[RESULT]```
105
+
106
+ ### Relative Grading (Pairwise Ranking)
107
+ Prometheus requires 4 components in the input: An instruction, 2 responses to evaluate, a score rubric, and a reference answer. You could refer to the prompt format below.
108
+ You should fill in the instruction, 2 responses, reference answer, and criteria description.
109
+
110
+ Fix the components with \{text\} inside.
111
+ ```
112
+ ###Task Description:
113
+ An instruction (might include an Input inside it), two responses to evaluate (denoted as Response A and Response B), a reference answer, and an evaluation criteria are given.
114
+ 1. Write a detailed feedback that assess the quality of the two responses strictly based on the given evaluation criteria, not evaluating in general.
115
+ 2. Make comparisons between Response A, Response B, and the Reference Answer. Instead of examining Response A and Response B separately, go straight to the point and mention about the commonalities and differences between them.
116
+ 3. After writing the feedback, indicate the better response, either "A" or "B".
117
+ 4. The output format should look as follows: "Feedback: (write a feedback for criteria) [RESULT] (Either "A" or "B")"
118
+ 5. Please do not generate any other opening, closing, and explanations.
119
+
120
+ ###Instruction:
121
+ {orig_instruction}
122
+
123
+ ###Response A:
124
+ {orig_response_A}
125
+
126
+ ###Response B:
127
+ {orig_response_B}
128
+
129
+ ###Reference Answer:
130
+ {orig_reference_answer}
131
+
132
+ ###Score Rubric:
133
+ {orig_criteria}
134
+
135
+ ###Feedback:
136
+ ```
137
+
138
+ After this, you should apply the conversation template of Mistral (not applying it might lead to unexpected behaviors).
139
+ You can find the conversation class at this [link](https://github.com/lm-sys/FastChat/blob/main/fastchat/conversation.py).
140
+ ```
141
+ conv = get_conv_template("mistral")
142
+ conv.set_system_message("You are a fair judge assistant assigned to deliver insightful feedback that compares individual performances, highlighting how each stands relative to others within the same cohort.")
143
+ conv.append_message(conv.roles[0], dialogs['instruction'])
144
+ conv.append_message(conv.roles[1], None)
145
+ prompt = conv.get_prompt()
146
+
147
+ x = tokenizer(prompt,truncation=False)
148
+ ```
149
+
150
+ As a result, a feedback and score decision will be generated, divided by a separating phrase ```[RESULT]```
151
+
152
+ ## License
153
+ Feedback Collection, Preference Collection, and Prometheus 2 are subject to OpenAI's Terms of Use for the generated data. If you suspect any violations, please reach out to us.
154
+
155
+
156
+ # Citation
157
+
158
+
159
+ If you find the following model helpful, please consider citing our paper!
160
+
161
+ **BibTeX:**
162
+
163
+ ```bibtex
164
+ @misc{kim2023prometheus,
165
+ title={Prometheus: Inducing Fine-grained Evaluation Capability in Language Models},
166
+ author={Seungone Kim and Jamin Shin and Yejin Cho and Joel Jang and Shayne Longpre and Hwaran Lee and Sangdoo Yun and Seongjin Shin and Sungdong Kim and James Thorne and Minjoon Seo},
167
+ year={2023},
168
+ eprint={2310.08491},
169
+ archivePrefix={arXiv},
170
+ primaryClass={cs.CL}
171
+ }
172
+ ```
173
+ ```bibtex
174
+ @misc{kim2024prometheus,
175
+ title={Prometheus 2: An Open Source Language Model Specialized in Evaluating Other Language Models},
176
+ author={Seungone Kim and Juyoung Suk and Shayne Longpre and Bill Yuchen Lin and Jamin Shin and Sean Welleck and Graham Neubig and Moontae Lee and Kyungjae Lee and Minjoon Seo},
177
+ year={2024},
178
+ eprint={2405.01535},
179
+ archivePrefix={arXiv},
180
+ primaryClass={cs.CL}
181
+ }
182
+ ```
config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "kaist-ai/prometheus-7b-v1.5-beta-1",
3
+ "architectures": [
4
+ "MistralForCausalLM"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 1,
8
+ "eos_token_id": 2,
9
+ "hidden_act": "silu",
10
+ "hidden_size": 4096,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 14336,
13
+ "max_position_embeddings": 32768,
14
+ "model_type": "mistral",
15
+ "num_attention_heads": 32,
16
+ "num_hidden_layers": 32,
17
+ "num_key_value_heads": 8,
18
+ "rms_norm_eps": 1e-05,
19
+ "rope_theta": 1000000.0,
20
+ "sliding_window": null,
21
+ "tie_word_embeddings": false,
22
+ "torch_dtype": "bfloat16",
23
+ "transformers_version": "4.35.2",
24
+ "use_cache": true,
25
+ "vocab_size": 32000
26
+ }
finegrained_eval.JPG ADDED
mergekit_config.yml ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ models:
3
+ - model: kaist-ai/prometheus-7b-v1.5-beta-1
4
+ parameters:
5
+ weight: 1.0
6
+ - model: kaist-ai/prometheus-7b-v1.5-beta-2
7
+ parameters:
8
+ weight: 1.0
9
+ merge_method: linear
10
+ dtype: bfloat16
model-00001-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:26afbe36c7d04dd26488eff13c5b7975f2e2a658eb6f3517c3f58e3c61b2ba2d
3
+ size 1979773128
model-00002-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68754ee9a08370759db5ccfa647915f3c6dca186d587a34b8223a32ba6723d54
3
+ size 1946235640
model-00003-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7044141b330ff94fd52f1fb18799156071ea212b045050f9f66b16f010867a58
3
+ size 1973490216
model-00004-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6f73b31c39c598501ebb1c9237c3efcac3798ea2b77c07e81b5f1d5a8a82f56
3
+ size 1979781464
model-00005-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1ff853ff8692412e3bc26acccc0c0b3430b8a11c531fcbd3a811f7410be2679c
3
+ size 1946243984
model-00006-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c3bedf39d42e46871e188bc50003f28a6100c82109c822b9cdeab3fc3468db45
3
+ size 1923166040
model-00007-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:96db009cb729bdec84a6df74d4f7413479f4ab98ff8df00a75ebbe9a93ae05de
3
+ size 1946243984
model-00008-of-00008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:714069c1ba0aa520a27a89ffc45cf1bc2af0adfaf3b91deaed5b7e95b51fbf06
3
+ size 788563544
model.safetensors.index.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"metadata": {"mergekit_version": "0.0.4.1"}, "weight_map": {"model.layers.10.mlp.gate_proj.weight": "model-00001-of-00008.safetensors", "model.layers.10.mlp.up_proj.weight": "model-00001-of-00008.safetensors", "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00008.safetensors", "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00008.safetensors", "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00008.safetensors", "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00008.safetensors", "model.layers.9.mlp.down_proj.weight": "model-00001-of-00008.safetensors", "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00008.safetensors", "model.layers.9.mlp.up_proj.weight": "model-00001-of-00008.safetensors", "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00008.safetensors", "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00008.safetensors", "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00008.safetensors", "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00008.safetensors", "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00008.safetensors", "model.layers.9.input_layernorm.weight": "model-00001-of-00008.safetensors", "model.layers.8.mlp.down_proj.weight": "model-00001-of-00008.safetensors", "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00008.safetensors", "model.layers.8.mlp.up_proj.weight": "model-00001-of-00008.safetensors", "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00008.safetensors", "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00008.safetensors", "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00008.safetensors", "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00008.safetensors", "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00008.safetensors", "model.layers.8.input_layernorm.weight": "model-00001-of-00008.safetensors", "model.layers.7.mlp.down_proj.weight": "model-00001-of-00008.safetensors", "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00008.safetensors", "model.layers.7.mlp.up_proj.weight": "model-00001-of-00008.safetensors", "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00008.safetensors", "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00008.safetensors", "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00008.safetensors", "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00008.safetensors", "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00008.safetensors", "model.layers.7.input_layernorm.weight": "model-00001-of-00008.safetensors", "model.layers.6.mlp.down_proj.weight": "model-00001-of-00008.safetensors", "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00008.safetensors", "model.layers.6.mlp.up_proj.weight": "model-00001-of-00008.safetensors", "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00008.safetensors", "model.layers.6.self_attn.o_proj.weight": "model-00002-of-00008.safetensors", "model.layers.6.self_attn.v_proj.weight": "model-00002-of-00008.safetensors", "model.layers.6.self_attn.k_proj.weight": "model-00002-of-00008.safetensors", "model.layers.6.self_attn.q_proj.weight": "model-00002-of-00008.safetensors", "model.layers.6.input_layernorm.weight": "model-00002-of-00008.safetensors", "model.layers.5.mlp.down_proj.weight": "model-00002-of-00008.safetensors", "model.layers.5.mlp.gate_proj.weight": "model-00002-of-00008.safetensors", "model.layers.5.mlp.up_proj.weight": "model-00002-of-00008.safetensors", "model.layers.5.post_attention_layernorm.weight": "model-00002-of-00008.safetensors", "model.layers.5.self_attn.o_proj.weight": "model-00002-of-00008.safetensors", "model.layers.5.self_attn.v_proj.weight": "model-00002-of-00008.safetensors", "model.layers.5.self_attn.k_proj.weight": "model-00002-of-00008.safetensors", "model.layers.5.self_attn.q_proj.weight": "model-00002-of-00008.safetensors", "model.layers.5.input_layernorm.weight": "model-00002-of-00008.safetensors", "model.layers.4.mlp.down_proj.weight": "model-00002-of-00008.safetensors", "model.layers.4.mlp.gate_proj.weight": "model-00002-of-00008.safetensors", "model.layers.4.mlp.up_proj.weight": "model-00002-of-00008.safetensors", "model.layers.4.post_attention_layernorm.weight": "model-00002-of-00008.safetensors", "model.layers.4.self_attn.o_proj.weight": "model-00002-of-00008.safetensors", "model.layers.4.self_attn.v_proj.weight": "model-00002-of-00008.safetensors", "model.layers.4.self_attn.k_proj.weight": "model-00002-of-00008.safetensors", "model.layers.4.self_attn.q_proj.weight": "model-00002-of-00008.safetensors", "model.layers.4.input_layernorm.weight": "model-00002-of-00008.safetensors", "model.layers.3.mlp.down_proj.weight": "model-00002-of-00008.safetensors", "model.layers.3.mlp.gate_proj.weight": "model-00002-of-00008.safetensors", "model.layers.3.mlp.up_proj.weight": "model-00002-of-00008.safetensors", "model.layers.3.post_attention_layernorm.weight": "model-00002-of-00008.safetensors", "model.layers.3.self_attn.o_proj.weight": "model-00002-of-00008.safetensors", "model.layers.3.self_attn.v_proj.weight": "model-00002-of-00008.safetensors", "model.layers.3.self_attn.k_proj.weight": "model-00002-of-00008.safetensors", "model.layers.3.self_attn.q_proj.weight": "model-00002-of-00008.safetensors", "model.layers.3.input_layernorm.weight": "model-00002-of-00008.safetensors", "model.layers.2.mlp.down_proj.weight": "model-00002-of-00008.safetensors", "model.layers.2.mlp.gate_proj.weight": "model-00002-of-00008.safetensors", "model.layers.2.mlp.up_proj.weight": "model-00002-of-00008.safetensors", "model.layers.2.post_attention_layernorm.weight": "model-00002-of-00008.safetensors", "model.layers.2.self_attn.o_proj.weight": "model-00002-of-00008.safetensors", "model.layers.2.self_attn.v_proj.weight": "model-00002-of-00008.safetensors", "model.layers.2.self_attn.k_proj.weight": "model-00002-of-00008.safetensors", "model.layers.2.self_attn.q_proj.weight": "model-00002-of-00008.safetensors", "model.layers.2.input_layernorm.weight": "model-00002-of-00008.safetensors", "model.layers.1.mlp.down_proj.weight": "model-00002-of-00008.safetensors", "model.layers.1.mlp.gate_proj.weight": "model-00003-of-00008.safetensors", "model.layers.1.mlp.up_proj.weight": "model-00003-of-00008.safetensors", "model.layers.1.post_attention_layernorm.weight": "model-00003-of-00008.safetensors", "model.layers.1.self_attn.o_proj.weight": "model-00003-of-00008.safetensors", "model.layers.1.self_attn.v_proj.weight": "model-00003-of-00008.safetensors", "model.layers.1.self_attn.k_proj.weight": "model-00003-of-00008.safetensors", "model.layers.1.self_attn.q_proj.weight": "model-00003-of-00008.safetensors", "model.layers.1.input_layernorm.weight": "model-00003-of-00008.safetensors", "model.layers.0.mlp.down_proj.weight": "model-00003-of-00008.safetensors", "model.layers.0.mlp.gate_proj.weight": "model-00003-of-00008.safetensors", "model.layers.0.mlp.up_proj.weight": "model-00003-of-00008.safetensors", "model.layers.0.post_attention_layernorm.weight": "model-00003-of-00008.safetensors", "model.layers.0.self_attn.o_proj.weight": "model-00003-of-00008.safetensors", "model.layers.0.self_attn.v_proj.weight": "model-00003-of-00008.safetensors", "model.layers.0.self_attn.k_proj.weight": "model-00003-of-00008.safetensors", "model.layers.0.self_attn.q_proj.weight": "model-00003-of-00008.safetensors", "model.layers.0.input_layernorm.weight": "model-00003-of-00008.safetensors", "model.embed_tokens.weight": "model-00003-of-00008.safetensors", "model.layers.22.self_attn.o_proj.weight": "model-00003-of-00008.safetensors", "model.layers.22.self_attn.v_proj.weight": "model-00003-of-00008.safetensors", "model.layers.22.self_attn.k_proj.weight": "model-00003-of-00008.safetensors", "model.layers.22.self_attn.q_proj.weight": "model-00003-of-00008.safetensors", "model.layers.21.mlp.down_proj.weight": "model-00003-of-00008.safetensors", "model.layers.21.mlp.gate_proj.weight": "model-00003-of-00008.safetensors", "model.layers.21.mlp.up_proj.weight": "model-00003-of-00008.safetensors", "model.layers.21.post_attention_layernorm.weight": "model-00003-of-00008.safetensors", "model.layers.21.self_attn.o_proj.weight": "model-00003-of-00008.safetensors", "model.layers.21.self_attn.v_proj.weight": "model-00003-of-00008.safetensors", "model.layers.21.self_attn.k_proj.weight": "model-00003-of-00008.safetensors", "model.layers.21.self_attn.q_proj.weight": "model-00003-of-00008.safetensors", "model.layers.21.input_layernorm.weight": "model-00003-of-00008.safetensors", "model.layers.20.mlp.down_proj.weight": "model-00003-of-00008.safetensors", "model.layers.20.mlp.gate_proj.weight": "model-00003-of-00008.safetensors", "model.layers.20.mlp.up_proj.weight": "model-00003-of-00008.safetensors", "model.layers.20.post_attention_layernorm.weight": "model-00003-of-00008.safetensors", "model.layers.20.self_attn.o_proj.weight": "model-00003-of-00008.safetensors", "model.layers.20.self_attn.v_proj.weight": "model-00003-of-00008.safetensors", "model.layers.20.self_attn.k_proj.weight": "model-00003-of-00008.safetensors", "model.layers.20.self_attn.q_proj.weight": "model-00003-of-00008.safetensors", "model.layers.20.input_layernorm.weight": "model-00003-of-00008.safetensors", "model.layers.19.mlp.down_proj.weight": "model-00004-of-00008.safetensors", "model.layers.19.mlp.gate_proj.weight": "model-00004-of-00008.safetensors", "model.layers.19.mlp.up_proj.weight": "model-00004-of-00008.safetensors", "model.layers.19.post_attention_layernorm.weight": "model-00004-of-00008.safetensors", "model.layers.19.self_attn.o_proj.weight": "model-00004-of-00008.safetensors", "model.layers.19.self_attn.v_proj.weight": "model-00004-of-00008.safetensors", "model.layers.19.self_attn.k_proj.weight": "model-00004-of-00008.safetensors", "model.layers.19.self_attn.q_proj.weight": "model-00004-of-00008.safetensors", "model.layers.19.input_layernorm.weight": "model-00004-of-00008.safetensors", "model.layers.18.mlp.down_proj.weight": "model-00004-of-00008.safetensors", "model.layers.18.mlp.gate_proj.weight": "model-00004-of-00008.safetensors", "model.layers.18.mlp.up_proj.weight": "model-00004-of-00008.safetensors", "model.layers.18.post_attention_layernorm.weight": "model-00004-of-00008.safetensors", "model.layers.18.self_attn.o_proj.weight": "model-00004-of-00008.safetensors", "model.layers.18.self_attn.v_proj.weight": "model-00004-of-00008.safetensors", "model.layers.18.self_attn.k_proj.weight": "model-00004-of-00008.safetensors", "model.layers.18.self_attn.q_proj.weight": "model-00004-of-00008.safetensors", "model.layers.18.input_layernorm.weight": "model-00004-of-00008.safetensors", "model.layers.17.mlp.down_proj.weight": "model-00004-of-00008.safetensors", "model.layers.17.mlp.gate_proj.weight": "model-00004-of-00008.safetensors", "model.layers.17.mlp.up_proj.weight": "model-00004-of-00008.safetensors", "model.layers.17.post_attention_layernorm.weight": "model-00004-of-00008.safetensors", "model.layers.17.self_attn.o_proj.weight": "model-00004-of-00008.safetensors", "model.layers.17.self_attn.v_proj.weight": "model-00004-of-00008.safetensors", "model.layers.17.self_attn.k_proj.weight": "model-00004-of-00008.safetensors", "model.layers.17.self_attn.q_proj.weight": "model-00004-of-00008.safetensors", "model.layers.17.input_layernorm.weight": "model-00004-of-00008.safetensors", "model.layers.16.mlp.down_proj.weight": "model-00004-of-00008.safetensors", "model.layers.16.mlp.gate_proj.weight": "model-00004-of-00008.safetensors", "model.layers.16.mlp.up_proj.weight": "model-00004-of-00008.safetensors", "model.layers.16.post_attention_layernorm.weight": "model-00004-of-00008.safetensors", "model.layers.16.self_attn.o_proj.weight": "model-00004-of-00008.safetensors", "model.layers.16.self_attn.v_proj.weight": "model-00004-of-00008.safetensors", "model.layers.16.self_attn.k_proj.weight": "model-00004-of-00008.safetensors", "model.layers.16.self_attn.q_proj.weight": "model-00004-of-00008.safetensors", "model.layers.16.input_layernorm.weight": "model-00004-of-00008.safetensors", "model.layers.15.mlp.down_proj.weight": "model-00004-of-00008.safetensors", "model.layers.15.mlp.gate_proj.weight": "model-00004-of-00008.safetensors", "model.layers.15.mlp.up_proj.weight": "model-00005-of-00008.safetensors", "model.layers.15.post_attention_layernorm.weight": "model-00005-of-00008.safetensors", "model.layers.15.self_attn.o_proj.weight": "model-00005-of-00008.safetensors", "model.layers.15.self_attn.v_proj.weight": "model-00005-of-00008.safetensors", "model.layers.15.self_attn.k_proj.weight": "model-00005-of-00008.safetensors", "model.layers.15.self_attn.q_proj.weight": "model-00005-of-00008.safetensors", "model.layers.15.input_layernorm.weight": "model-00005-of-00008.safetensors", "model.layers.14.mlp.down_proj.weight": "model-00005-of-00008.safetensors", "model.layers.14.mlp.gate_proj.weight": "model-00005-of-00008.safetensors", "model.layers.14.mlp.up_proj.weight": "model-00005-of-00008.safetensors", "model.layers.14.post_attention_layernorm.weight": "model-00005-of-00008.safetensors", "model.layers.14.self_attn.o_proj.weight": "model-00005-of-00008.safetensors", "model.layers.14.self_attn.v_proj.weight": "model-00005-of-00008.safetensors", "model.layers.14.self_attn.k_proj.weight": "model-00005-of-00008.safetensors", "model.layers.14.self_attn.q_proj.weight": "model-00005-of-00008.safetensors", "model.layers.14.input_layernorm.weight": "model-00005-of-00008.safetensors", "model.layers.13.mlp.down_proj.weight": "model-00005-of-00008.safetensors", "model.layers.13.mlp.gate_proj.weight": "model-00005-of-00008.safetensors", "model.layers.13.mlp.up_proj.weight": "model-00005-of-00008.safetensors", "model.layers.13.post_attention_layernorm.weight": "model-00005-of-00008.safetensors", "model.layers.13.self_attn.o_proj.weight": "model-00005-of-00008.safetensors", "model.layers.13.self_attn.v_proj.weight": "model-00005-of-00008.safetensors", "model.layers.13.self_attn.k_proj.weight": "model-00005-of-00008.safetensors", "model.layers.13.self_attn.q_proj.weight": "model-00005-of-00008.safetensors", "model.layers.13.input_layernorm.weight": "model-00005-of-00008.safetensors", "model.layers.12.mlp.down_proj.weight": "model-00005-of-00008.safetensors", "model.layers.12.mlp.gate_proj.weight": "model-00005-of-00008.safetensors", "model.layers.12.mlp.up_proj.weight": "model-00005-of-00008.safetensors", "model.layers.12.post_attention_layernorm.weight": "model-00005-of-00008.safetensors", "model.layers.12.self_attn.o_proj.weight": "model-00005-of-00008.safetensors", "model.layers.12.self_attn.v_proj.weight": "model-00005-of-00008.safetensors", "model.layers.12.self_attn.k_proj.weight": "model-00005-of-00008.safetensors", "model.layers.12.self_attn.q_proj.weight": "model-00005-of-00008.safetensors", "model.layers.12.input_layernorm.weight": "model-00005-of-00008.safetensors", "model.layers.11.mlp.down_proj.weight": "model-00005-of-00008.safetensors", "model.layers.11.mlp.gate_proj.weight": "model-00005-of-00008.safetensors", "model.layers.11.mlp.up_proj.weight": "model-00005-of-00008.safetensors", "model.layers.11.post_attention_layernorm.weight": "model-00005-of-00008.safetensors", "model.layers.11.self_attn.o_proj.weight": "model-00005-of-00008.safetensors", "model.layers.11.self_attn.v_proj.weight": "model-00005-of-00008.safetensors", "model.layers.11.self_attn.k_proj.weight": "model-00005-of-00008.safetensors", "model.layers.11.self_attn.q_proj.weight": "model-00005-of-00008.safetensors", "model.layers.11.input_layernorm.weight": "model-00005-of-00008.safetensors", "model.layers.10.mlp.down_proj.weight": "model-00006-of-00008.safetensors", "model.layers.10.post_attention_layernorm.weight": "model-00006-of-00008.safetensors", "model.layers.10.input_layernorm.weight": "model-00006-of-00008.safetensors", "lm_head.weight": "model-00006-of-00008.safetensors", "model.norm.weight": "model-00006-of-00008.safetensors", "model.layers.31.mlp.down_proj.weight": "model-00006-of-00008.safetensors", "model.layers.31.mlp.gate_proj.weight": "model-00006-of-00008.safetensors", "model.layers.31.mlp.up_proj.weight": "model-00006-of-00008.safetensors", "model.layers.31.post_attention_layernorm.weight": "model-00006-of-00008.safetensors", "model.layers.31.self_attn.o_proj.weight": "model-00006-of-00008.safetensors", "model.layers.31.self_attn.v_proj.weight": "model-00006-of-00008.safetensors", "model.layers.31.self_attn.k_proj.weight": "model-00006-of-00008.safetensors", "model.layers.31.self_attn.q_proj.weight": "model-00006-of-00008.safetensors", "model.layers.31.input_layernorm.weight": "model-00006-of-00008.safetensors", "model.layers.30.mlp.down_proj.weight": "model-00006-of-00008.safetensors", "model.layers.30.mlp.gate_proj.weight": "model-00006-of-00008.safetensors", "model.layers.30.mlp.up_proj.weight": "model-00006-of-00008.safetensors", "model.layers.30.post_attention_layernorm.weight": "model-00006-of-00008.safetensors", "model.layers.30.self_attn.o_proj.weight": "model-00006-of-00008.safetensors", "model.layers.30.self_attn.v_proj.weight": "model-00006-of-00008.safetensors", "model.layers.30.self_attn.k_proj.weight": "model-00006-of-00008.safetensors", "model.layers.30.self_attn.q_proj.weight": "model-00006-of-00008.safetensors", "model.layers.30.input_layernorm.weight": "model-00006-of-00008.safetensors", "model.layers.29.mlp.down_proj.weight": "model-00006-of-00008.safetensors", "model.layers.29.mlp.gate_proj.weight": "model-00006-of-00008.safetensors", "model.layers.29.mlp.up_proj.weight": "model-00006-of-00008.safetensors", "model.layers.29.post_attention_layernorm.weight": "model-00006-of-00008.safetensors", "model.layers.29.self_attn.o_proj.weight": "model-00006-of-00008.safetensors", "model.layers.29.self_attn.v_proj.weight": "model-00006-of-00008.safetensors", "model.layers.29.self_attn.k_proj.weight": "model-00006-of-00008.safetensors", "model.layers.29.self_attn.q_proj.weight": "model-00006-of-00008.safetensors", "model.layers.29.input_layernorm.weight": "model-00006-of-00008.safetensors", "model.layers.28.mlp.down_proj.weight": "model-00006-of-00008.safetensors", "model.layers.28.mlp.gate_proj.weight": "model-00006-of-00008.safetensors", "model.layers.28.mlp.up_proj.weight": "model-00007-of-00008.safetensors", "model.layers.28.post_attention_layernorm.weight": "model-00007-of-00008.safetensors", "model.layers.28.self_attn.o_proj.weight": "model-00007-of-00008.safetensors", "model.layers.28.self_attn.v_proj.weight": "model-00007-of-00008.safetensors", "model.layers.28.self_attn.k_proj.weight": "model-00007-of-00008.safetensors", "model.layers.28.self_attn.q_proj.weight": "model-00007-of-00008.safetensors", "model.layers.28.input_layernorm.weight": "model-00007-of-00008.safetensors", "model.layers.27.mlp.down_proj.weight": "model-00007-of-00008.safetensors", "model.layers.27.mlp.gate_proj.weight": "model-00007-of-00008.safetensors", "model.layers.27.mlp.up_proj.weight": "model-00007-of-00008.safetensors", "model.layers.27.post_attention_layernorm.weight": "model-00007-of-00008.safetensors", "model.layers.27.self_attn.o_proj.weight": "model-00007-of-00008.safetensors", "model.layers.27.self_attn.v_proj.weight": "model-00007-of-00008.safetensors", "model.layers.27.self_attn.k_proj.weight": "model-00007-of-00008.safetensors", "model.layers.27.self_attn.q_proj.weight": "model-00007-of-00008.safetensors", "model.layers.27.input_layernorm.weight": "model-00007-of-00008.safetensors", "model.layers.26.mlp.down_proj.weight": "model-00007-of-00008.safetensors", "model.layers.26.mlp.gate_proj.weight": "model-00007-of-00008.safetensors", "model.layers.26.mlp.up_proj.weight": "model-00007-of-00008.safetensors", "model.layers.26.post_attention_layernorm.weight": "model-00007-of-00008.safetensors", "model.layers.26.self_attn.o_proj.weight": "model-00007-of-00008.safetensors", "model.layers.26.self_attn.v_proj.weight": "model-00007-of-00008.safetensors", "model.layers.26.self_attn.k_proj.weight": "model-00007-of-00008.safetensors", "model.layers.26.self_attn.q_proj.weight": "model-00007-of-00008.safetensors", "model.layers.26.input_layernorm.weight": "model-00007-of-00008.safetensors", "model.layers.25.mlp.down_proj.weight": "model-00007-of-00008.safetensors", "model.layers.25.mlp.gate_proj.weight": "model-00007-of-00008.safetensors", "model.layers.25.mlp.up_proj.weight": "model-00007-of-00008.safetensors", "model.layers.25.post_attention_layernorm.weight": "model-00007-of-00008.safetensors", "model.layers.25.self_attn.o_proj.weight": "model-00007-of-00008.safetensors", "model.layers.25.self_attn.v_proj.weight": "model-00007-of-00008.safetensors", "model.layers.25.self_attn.k_proj.weight": "model-00007-of-00008.safetensors", "model.layers.25.self_attn.q_proj.weight": "model-00007-of-00008.safetensors", "model.layers.25.input_layernorm.weight": "model-00007-of-00008.safetensors", "model.layers.24.mlp.down_proj.weight": "model-00007-of-00008.safetensors", "model.layers.24.mlp.gate_proj.weight": "model-00007-of-00008.safetensors", "model.layers.24.mlp.up_proj.weight": "model-00007-of-00008.safetensors", "model.layers.24.post_attention_layernorm.weight": "model-00007-of-00008.safetensors", "model.layers.24.self_attn.o_proj.weight": "model-00007-of-00008.safetensors", "model.layers.24.self_attn.v_proj.weight": "model-00007-of-00008.safetensors", "model.layers.24.self_attn.k_proj.weight": "model-00007-of-00008.safetensors", "model.layers.24.self_attn.q_proj.weight": "model-00007-of-00008.safetensors", "model.layers.24.input_layernorm.weight": "model-00007-of-00008.safetensors", "model.layers.23.mlp.down_proj.weight": "model-00008-of-00008.safetensors", "model.layers.23.mlp.gate_proj.weight": "model-00008-of-00008.safetensors", "model.layers.23.mlp.up_proj.weight": "model-00008-of-00008.safetensors", "model.layers.23.post_attention_layernorm.weight": "model-00008-of-00008.safetensors", "model.layers.23.self_attn.o_proj.weight": "model-00008-of-00008.safetensors", "model.layers.23.self_attn.v_proj.weight": "model-00008-of-00008.safetensors", "model.layers.23.self_attn.k_proj.weight": "model-00008-of-00008.safetensors", "model.layers.23.self_attn.q_proj.weight": "model-00008-of-00008.safetensors", "model.layers.23.input_layernorm.weight": "model-00008-of-00008.safetensors", "model.layers.22.mlp.down_proj.weight": "model-00008-of-00008.safetensors", "model.layers.22.mlp.gate_proj.weight": "model-00008-of-00008.safetensors", "model.layers.22.mlp.up_proj.weight": "model-00008-of-00008.safetensors", "model.layers.22.post_attention_layernorm.weight": "model-00008-of-00008.safetensors", "model.layers.22.input_layernorm.weight": "model-00008-of-00008.safetensors"}}
special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "</s>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "<unk>",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dadfd56d766715c61d2ef780a525ab43b8e6da4de6865bda3d95fdef5e134055
3
+ size 493443
tokenizer_config.json ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "<unk>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "<s>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "</s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ }
27
+ },
28
+ "additional_special_tokens": [],
29
+ "bos_token": "<s>",
30
+ "chat_template": "{%- if messages[0]['role'] == 'system' %}\n {%- set system_message = messages[0]['content'] %}\n {%- set loop_messages = messages[1:] %}\n{%- else %}\n {%- set loop_messages = messages %}\n{%- endif %}\n\n{{- bos_token }}\n{%- for message in loop_messages %}\n {%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) %}\n {{- raise_exception('After the optional system message, conversation roles must alternate user/assistant/user/assistant/...') }}\n {%- endif %}\n {%- if message['role'] == 'user' %}\n {%- if loop.first and system_message is defined %}\n {{- ' [INST] ' + system_message + '\\n\\n' + message['content'] + ' [/INST]' }}\n {%- else %}\n {{- ' [INST] ' + message['content'] + ' [/INST]' }}\n {%- endif %}\n {%- elif message['role'] == 'assistant' %}\n {{- ' ' + message['content'] + eos_token}}\n {%- else %}\n {{- raise_exception('Only user and assistant roles are supported, with the exception of an initial optional system message!') }}\n {%- endif %}\n{%- endfor %}\n",
31
+ "clean_up_tokenization_spaces": false,
32
+ "eos_token": "</s>",
33
+ "legacy": true,
34
+ "max_length": 4096,
35
+ "model_max_length": 1000000000000000019884624838656,
36
+ "pad_token": "</s>",
37
+ "sp_model_kwargs": {},
38
+ "spaces_between_special_tokens": false,
39
+ "stride": 0,
40
+ "tokenizer_class": "LlamaTokenizer",
41
+ "truncation_side": "right",
42
+ "truncation_strategy": "longest_first",
43
+ "unk_token": "<unk>",
44
+ "use_default_system_prompt": false
45
+ }