sanchit-gandhi HF staff commited on
Commit
c4f32d0
1 Parent(s): 0343bfc

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +90 -0
README.md ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ This is a copy of Mistral's [7B-Instruct-v0.1](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.1) Large Language Model (LLM).
14
+ Compared to the original version, who's weights are stored as 10GB shards, this version stores the weights in 5GB shards. Using smaller shards permits the use of the model on a Google Colab Free Tier without exceeding the system RAM (10GB). In all other ways, this model is identical to the original Mistral model.
15
+
16
+ The model 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.
17
+
18
+ For full details of this model please read the [paper](https://arxiv.org/abs/2310.06825) and [release blog post](https://mistral.ai/news/announcing-mistral-7b/).
19
+
20
+ ## Instruction format
21
+
22
+ 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.
23
+
24
+ E.g.
25
+ ```
26
+ text = "<s>[INST] What is your favourite condiment? [/INST]"
27
+ "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> "
28
+ "[INST] Do you have mayonnaise recipes? [/INST]"
29
+ ```
30
+
31
+ This format is available as a [chat template](https://huggingface.co/docs/transformers/main/chat_templating) via the `apply_chat_template()` method:
32
+
33
+ ```python
34
+ from transformers import AutoModelForCausalLM, AutoTokenizer
35
+
36
+ device = "cuda" # the device to load the model onto
37
+
38
+ model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
39
+ tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
40
+
41
+ messages = [
42
+ {"role": "user", "content": "What is your favourite condiment?"},
43
+ {"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!"},
44
+ {"role": "user", "content": "Do you have mayonnaise recipes?"}
45
+ ]
46
+
47
+ encodeds = tokenizer.apply_chat_template(messages, return_tensors="pt")
48
+
49
+ model_inputs = encodeds.to(device)
50
+ model.to(device)
51
+
52
+ generated_ids = model.generate(model_inputs, max_new_tokens=1000, do_sample=True)
53
+ decoded = tokenizer.batch_decode(generated_ids)
54
+ print(decoded[0])
55
+ ```
56
+
57
+ ## Model Architecture
58
+ This instruction model is based on Mistral-7B-v0.1, a transformer model with the following architecture choices:
59
+ - Grouped-Query Attention
60
+ - Sliding-Window Attention
61
+ - Byte-fallback BPE tokenizer
62
+
63
+ ## Troubleshooting
64
+ - If you see the following error:
65
+ ```
66
+ Traceback (most recent call last):
67
+ File "", line 1, in
68
+ File "/transformers/models/auto/auto_factory.py", line 482, in from_pretrained
69
+ config, kwargs = AutoConfig.from_pretrained(
70
+ File "/transformers/models/auto/configuration_auto.py", line 1022, in from_pretrained
71
+ config_class = CONFIG_MAPPING[config_dict["model_type"]]
72
+ File "/transformers/models/auto/configuration_auto.py", line 723, in getitem
73
+ raise KeyError(key)
74
+ KeyError: 'mistral'
75
+ ```
76
+
77
+ Installing transformers from source should solve the issue
78
+ pip install git+https://github.com/huggingface/transformers
79
+
80
+ This should not be required after transformers-v4.33.4.
81
+
82
+ ## Limitations
83
+
84
+ The Mistral 7B Instruct model is a quick demonstration that the base model can be easily fine-tuned to achieve compelling performance.
85
+ It does not have any moderation mechanisms. We're looking forward to engaging with the community on ways to
86
+ make the model finely respect guardrails, allowing for deployment in environments requiring moderated outputs.
87
+
88
+ ## The Mistral AI Team
89
+
90
+ 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.