DogOnKeyboard commited on
Commit
59a45f2
·
verified ·
1 Parent(s): 6c832cd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +229 -3
README.md CHANGED
@@ -1,3 +1,229 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: vllm
3
+ license: apache-2.0
4
+ base_model: mistralai/Mistral-7B-Instruct-v0.3
5
+ extra_gated_description: If you want to learn more about how we process your personal
6
+ data, please read our <a href="https://mistral.ai/terms/">Privacy Policy</a>.
7
+ tags:
8
+ - vllm
9
+ - mistral-common
10
+ - heretic
11
+ - uncensored
12
+ - decensored
13
+ - abliterated
14
+ ---
15
+ # This is a decensored version of [mistralai/Mistral-7B-Instruct-v0.3](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.3), made using [Heretic](https://github.com/p-e-w/heretic) v1.0.1
16
+
17
+ ## Abliteration parameters
18
+
19
+ | Parameter | Value |
20
+ | :-------- | :---: |
21
+ | **direction_index** | 16.03 |
22
+ | **attn.o_proj.max_weight** | 1.01 |
23
+ | **attn.o_proj.max_weight_position** | 23.50 |
24
+ | **attn.o_proj.min_weight** | 0.01 |
25
+ | **attn.o_proj.min_weight_distance** | 1.81 |
26
+ | **mlp.down_proj.max_weight** | 0.88 |
27
+ | **mlp.down_proj.max_weight_position** | 29.20 |
28
+ | **mlp.down_proj.min_weight** | 0.41 |
29
+ | **mlp.down_proj.min_weight_distance** | 6.05 |
30
+
31
+
32
+ ## Performance
33
+
34
+ | Metric | This model | Original model ([mistralai/Mistral-7B-Instruct-v0.3](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.3)) |
35
+ | :----- | :--------: | :---------------------------: |
36
+ | **KL divergence** | 0.08 | 0 *(by definition)* |
37
+ | **Refusals** | 2/100 | 86/100 |
38
+
39
+ -----
40
+
41
+
42
+ # Model Card for Mistral-7B-Instruct-v0.3
43
+
44
+ The Mistral-7B-Instruct-v0.3 Large Language Model (LLM) is an instruct fine-tuned version of the Mistral-7B-v0.3.
45
+
46
+ Mistral-7B-v0.3 has the following changes compared to [Mistral-7B-v0.2](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2/edit/main/README.md)
47
+ - Extended vocabulary to 32768
48
+ - Supports v3 Tokenizer
49
+ - Supports function calling
50
+
51
+ ## Installation
52
+
53
+ It is recommended to use `mistralai/Mistral-7B-Instruct-v0.3` with [mistral-inference](https://github.com/mistralai/mistral-inference). For HF transformers code snippets, please keep scrolling.
54
+
55
+ ```
56
+ pip install mistral_inference
57
+ ```
58
+
59
+ ## Download
60
+
61
+ ```py
62
+ from huggingface_hub import snapshot_download
63
+ from pathlib import Path
64
+
65
+ mistral_models_path = Path.home().joinpath('mistral_models', '7B-Instruct-v0.3')
66
+ mistral_models_path.mkdir(parents=True, exist_ok=True)
67
+
68
+ snapshot_download(repo_id="mistralai/Mistral-7B-Instruct-v0.3", allow_patterns=["params.json", "consolidated.safetensors", "tokenizer.model.v3"], local_dir=mistral_models_path)
69
+ ```
70
+
71
+ ### Chat
72
+
73
+ After installing `mistral_inference`, a `mistral-chat` CLI command should be available in your environment. You can chat with the model using
74
+
75
+ ```
76
+ mistral-chat $HOME/mistral_models/7B-Instruct-v0.3 --instruct --max_tokens 256
77
+ ```
78
+
79
+ ### Instruct following
80
+
81
+ ```py
82
+ from mistral_inference.transformer import Transformer
83
+ from mistral_inference.generate import generate
84
+
85
+ from mistral_common.tokens.tokenizers.mistral import MistralTokenizer
86
+ from mistral_common.protocol.instruct.messages import UserMessage
87
+ from mistral_common.protocol.instruct.request import ChatCompletionRequest
88
+
89
+
90
+ tokenizer = MistralTokenizer.from_file(f"{mistral_models_path}/tokenizer.model.v3")
91
+ model = Transformer.from_folder(mistral_models_path)
92
+
93
+ completion_request = ChatCompletionRequest(messages=[UserMessage(content="Explain Machine Learning to me in a nutshell.")])
94
+
95
+ tokens = tokenizer.encode_chat_completion(completion_request).tokens
96
+
97
+ out_tokens, _ = generate([tokens], model, max_tokens=64, temperature=0.0, eos_id=tokenizer.instruct_tokenizer.tokenizer.eos_id)
98
+ result = tokenizer.instruct_tokenizer.tokenizer.decode(out_tokens[0])
99
+
100
+ print(result)
101
+ ```
102
+
103
+ ### Function calling
104
+
105
+ ```py
106
+ from mistral_common.protocol.instruct.tool_calls import Function, Tool
107
+ from mistral_inference.transformer import Transformer
108
+ from mistral_inference.generate import generate
109
+
110
+ from mistral_common.tokens.tokenizers.mistral import MistralTokenizer
111
+ from mistral_common.protocol.instruct.messages import UserMessage
112
+ from mistral_common.protocol.instruct.request import ChatCompletionRequest
113
+
114
+
115
+ tokenizer = MistralTokenizer.from_file(f"{mistral_models_path}/tokenizer.model.v3")
116
+ model = Transformer.from_folder(mistral_models_path)
117
+
118
+ completion_request = ChatCompletionRequest(
119
+ tools=[
120
+ Tool(
121
+ function=Function(
122
+ name="get_current_weather",
123
+ description="Get the current weather",
124
+ parameters={
125
+ "type": "object",
126
+ "properties": {
127
+ "location": {
128
+ "type": "string",
129
+ "description": "The city and state, e.g. San Francisco, CA",
130
+ },
131
+ "format": {
132
+ "type": "string",
133
+ "enum": ["celsius", "fahrenheit"],
134
+ "description": "The temperature unit to use. Infer this from the users location.",
135
+ },
136
+ },
137
+ "required": ["location", "format"],
138
+ },
139
+ )
140
+ )
141
+ ],
142
+ messages=[
143
+ UserMessage(content="What's the weather like today in Paris?"),
144
+ ],
145
+ )
146
+
147
+ tokens = tokenizer.encode_chat_completion(completion_request).tokens
148
+
149
+ out_tokens, _ = generate([tokens], model, max_tokens=64, temperature=0.0, eos_id=tokenizer.instruct_tokenizer.tokenizer.eos_id)
150
+ result = tokenizer.instruct_tokenizer.tokenizer.decode(out_tokens[0])
151
+
152
+ print(result)
153
+ ```
154
+
155
+ ## Generate with `transformers`
156
+
157
+ If you want to use Hugging Face `transformers` to generate text, you can do something like this.
158
+
159
+ ```py
160
+ from transformers import pipeline
161
+
162
+ messages = [
163
+ {"role": "system", "content": "You are a pirate chatbot who always responds in pirate speak!"},
164
+ {"role": "user", "content": "Who are you?"},
165
+ ]
166
+ chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
167
+ chatbot(messages)
168
+ ```
169
+
170
+
171
+ ## Function calling with `transformers`
172
+
173
+ To use this example, you'll need `transformers` version 4.42.0 or higher. Please see the
174
+ [function calling guide](https://huggingface.co/docs/transformers/main/chat_templating#advanced-tool-use--function-calling)
175
+ in the `transformers` docs for more information.
176
+
177
+ ```python
178
+ from transformers import AutoModelForCausalLM, AutoTokenizer
179
+ import torch
180
+
181
+ model_id = "mistralai/Mistral-7B-Instruct-v0.3"
182
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
183
+
184
+ def get_current_weather(location: str, format: str):
185
+ """
186
+ Get the current weather
187
+
188
+ Args:
189
+ location: The city and state, e.g. San Francisco, CA
190
+ format: The temperature unit to use. Infer this from the users location. (choices: ["celsius", "fahrenheit"])
191
+ """
192
+ pass
193
+
194
+ conversation = [{"role": "user", "content": "What's the weather like in Paris?"}]
195
+ tools = [get_current_weather]
196
+
197
+
198
+ # format and tokenize the tool use prompt
199
+ inputs = tokenizer.apply_chat_template(
200
+ conversation,
201
+ tools=tools,
202
+ add_generation_prompt=True,
203
+ return_dict=True,
204
+ return_tensors="pt",
205
+ )
206
+
207
+ model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
208
+
209
+ inputs.to(model.device)
210
+ outputs = model.generate(**inputs, max_new_tokens=1000)
211
+ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
212
+ ```
213
+
214
+ Note that, for reasons of space, this example does not show a complete cycle of calling a tool and adding the tool call and tool
215
+ results to the chat history so that the model can use them in its next generation. For a full tool calling example, please
216
+ see the [function calling guide](https://huggingface.co/docs/transformers/main/chat_templating#advanced-tool-use--function-calling),
217
+ and note that Mistral **does** use tool call IDs, so these must be included in your tool calls and tool results. They should be
218
+ exactly 9 alphanumeric characters.
219
+
220
+
221
+ ## Limitations
222
+
223
+ The Mistral 7B Instruct model is a quick demonstration that the base model can be easily fine-tuned to achieve compelling performance.
224
+ It does not have any moderation mechanisms. We're looking forward to engaging with the community on ways to
225
+ make the model finely respect guardrails, allowing for deployment in environments requiring moderated outputs.
226
+
227
+ ## The Mistral AI Team
228
+
229
+ Albert Jiang, Alexandre Sablayrolles, Alexis Tacnet, Antoine Roux, Arthur Mensch, Audrey Herblin-Stoop, Baptiste Bout, Baudouin de Monicault, Blanche Savary, Bam4d, Caroline Feldman, Devendra Singh Chaplot, Diego de las Casas, Eleonore Arcelin, Emma Bou Hanna, Etienne Metzger, Gianna Lengyel, Guillaume Bour, Guillaume Lample, Harizo Rajaona, Jean-Malo Delignon, Jia Li, Justus Murke, Louis Martin, Louis Ternon, Lucile Saulnier, Lélio Renard Lavaud, Margaret Jennings, Marie Pellat, Marie Torelli, Marie-Anne Lachaux, Nicolas Schuhl, Patrick von Platen, Pierre Stock, Sandeep Subramanian, Sophia Yang, Szymon Antoniak, Teven Le Scao, Thibaut Lavril, Timothée Lacroix, Théophile Gervet, Thomas Wang, Valera Nemychnikova, William El Sayed, William Marshall