perlthoughts commited on
Commit
aac8354
1 Parent(s): f72eea2

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +91 -0
README.md ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ pipeline_tag: text-generation
4
+ tags:
5
+ - finetuned
6
+ inference:
7
+ parameters:
8
+ temperature: 0.7
9
+ ---
10
+
11
+ # Model Card for Mistral-7B-Instruct-v0.1
12
+
13
+ # Extended to 16k Context Length
14
+
15
+ # Original Model Card
16
+
17
+ The Mistral-7B-Instruct-v0.1 Large Language Model (LLM) is a instruct fine-tuned version of the [Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) generative text model using a variety of publicly available conversation datasets.
18
+
19
+ For full details of this model please read our [paper](https://arxiv.org/abs/2310.06825) and [release blog post](https://mistral.ai/news/announcing-mistral-7b/).
20
+
21
+ ## Instruction format
22
+
23
+ In order to leverage instruction fine-tuning, your prompt should be surrounded by `[INST]` and `[/INST]` tokens. The very first instruction should begin with a begin of sentence id. The next instructions should not. The assistant generation will be ended by the end-of-sentence token id.
24
+
25
+ E.g.
26
+ ```
27
+ text = "<s>[INST] What is your favourite condiment? [/INST]"
28
+ "Well, I'm quite partial to a good squeeze of fresh lemon juice. It adds just the right amount of zesty flavour to whatever I'm cooking up in the kitchen!</s> "
29
+ "[INST] Do you have mayonnaise recipes? [/INST]"
30
+ ```
31
+
32
+ This format is available as a [chat template](https://huggingface.co/docs/transformers/main/chat_templating) via the `apply_chat_template()` method:
33
+
34
+ ```python
35
+ from transformers import AutoModelForCausalLM, AutoTokenizer
36
+
37
+ device = "cuda" # the device to load the model onto
38
+
39
+ model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
40
+ tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
41
+
42
+ messages = [
43
+ {"role": "user", "content": "What is your favourite condiment?"},
44
+ {"role": "assistant", "content": "Well, I'm quite partial to a good squeeze of fresh lemon juice. It adds just the right amount of zesty flavour to whatever I'm cooking up in the kitchen!"},
45
+ {"role": "user", "content": "Do you have mayonnaise recipes?"}
46
+ ]
47
+
48
+ encodeds = tokenizer.apply_chat_template(messages, return_tensors="pt")
49
+
50
+ model_inputs = encodeds.to(device)
51
+ model.to(device)
52
+
53
+ generated_ids = model.generate(model_inputs, max_new_tokens=1000, do_sample=True)
54
+ decoded = tokenizer.batch_decode(generated_ids)
55
+ print(decoded[0])
56
+ ```
57
+
58
+ ## Model Architecture
59
+ This instruction model is based on Mistral-7B-v0.1, a transformer model with the following architecture choices:
60
+ - Grouped-Query Attention
61
+ - Sliding-Window Attention
62
+ - Byte-fallback BPE tokenizer
63
+
64
+ ## Troubleshooting
65
+ - If you see the following error:
66
+ ```
67
+ Traceback (most recent call last):
68
+ File "", line 1, in
69
+ File "/transformers/models/auto/auto_factory.py", line 482, in from_pretrained
70
+ config, kwargs = AutoConfig.from_pretrained(
71
+ File "/transformers/models/auto/configuration_auto.py", line 1022, in from_pretrained
72
+ config_class = CONFIG_MAPPING[config_dict["model_type"]]
73
+ File "/transformers/models/auto/configuration_auto.py", line 723, in getitem
74
+ raise KeyError(key)
75
+ KeyError: 'mistral'
76
+ ```
77
+
78
+ Installing transformers from source should solve the issue
79
+ pip install git+https://github.com/huggingface/transformers
80
+
81
+ This should not be required after transformers-v4.33.4.
82
+
83
+ ## Limitations
84
+
85
+ The Mistral 7B Instruct model is a quick demonstration that the base model can be easily fine-tuned to achieve compelling performance.
86
+ It does not have any moderation mechanisms. We're looking forward to engaging with the community on ways to
87
+ make the model finely respect guardrails, allowing for deployment in environments requiring moderated outputs.
88
+
89
+ ## The Mistral AI Team
90
+
91
+ Albert Jiang, Alexandre Sablayrolles, Arthur Mensch, Chris Bamford, Devendra Singh Chaplot, Diego de las Casas, Florian Bressand, Gianna Lengyel, Guillaume Lample, Lélio Renard Lavaud, Lucile Saulnier, Marie-Anne Lachaux, Pierre Stock, Teven Le Scao, Thibaut Lavril, Thomas Wang, Timothée Lacroix, William El Sayed.