Update README.md

README.md

@@ -1,40 +1,46 @@
 ---
+license: apache-2.0
+pipeline_tag: text-generation
+language:
+  - en
 tags:
-- autotrain
-- text-generation
-widget:
-- text: "I love AutoTrain because "
-license: other
+- pretrained
+inference:
+  parameters:
+    temperature: 0.7
 ---
 
-# Model Trained Using AutoTrain
+# Model Card for Mistral-7B-v0.1
 
-This model was trained using AutoTrain. For more information, please visit [AutoTrain](https://hf.co/docs/autotrain).
+The Mistral-7B-v0.1 Large Language Model (LLM) is a pretrained generative text model with 7 billion parameters. 
+Mistral-7B-v0.1 outperforms Llama 2 13B on all benchmarks we tested.
 
-# Usage
+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/).
 
-```python
+## Model Architecture
 
-from transformers import AutoModelForCausalLM, AutoTokenizer
+Mistral-7B-v0.1 is a transformer model, with the following architecture choices:
+- Grouped-Query Attention
+- Sliding-Window Attention
+- Byte-fallback BPE tokenizer
 
-model_path = "PATH_TO_THIS_REPO"
+## Troubleshooting
 
-tokenizer = AutoTokenizer.from_pretrained(model_path)
-model = AutoModelForCausalLM.from_pretrained(
-    model_path,
-    device_map="auto",
-    torch_dtype='auto'
-).eval()
+- If you see the following error:
+```
+KeyError: 'mistral'
+```
+- Or:
+```
+NotImplementedError: Cannot copy out of meta tensor; no data!
+```
 
-# Prompt content: "hi"
-messages = [
-    {"role": "user", "content": "hi"}
-]
+Ensure you are utilizing a stable version of Transformers, 4.34.0 or newer.
 
-input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')
-output_ids = model.generate(input_ids.to('cuda'))
-response = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
+## Notice
 
-# Model response: "Hello! How can I assist you today?"
-print(response)
-```
+Mistral 7B is a pretrained base model and therefore does not have any moderation mechanisms.
+
+## The Mistral AI Team
+ 
+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.