Update README.md

README.md

@@ -21,11 +21,20 @@ To use, pass `trust_remote_code=True` when loading the model, for example
 
 ```python
 model = AutoModelForCausalLM.from_pretrained("NousResearch/Yarn-Mistral-7b-64k",
+  use_flash_attention_2=True,
   torch_dtype=torch.bfloat16,
   device_map="auto",
   trust_remote_code=True)
 ```
 
+## Benchmarks
+
+| Model | Context Window | ARC-c | Hellaswag | MMLU | Truthful QA |
+|-------|---------------:|------:|----------:|-----:|------------:|
+| [Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) | 8K | 59.98 | 83.31 | 64.16 | 42.15 |
+| [Yarn-Mistral-7b-64k](https://huggingface.co/NousResearch/Yarn-Mistral-7b-64k) | 64K | 59.38 | 81.21 | 61.32 | 42.50 |
+| [Yarn-Mistral-7b-128k](https://huggingface.co/NousResearch/Yarn-Mistral-7b-128k) | 128K | 58.87 | 80.58 | 60.64 | 42.46 |
+
 ## Collaborators
 
  - [bloc97](https://github.com/bloc97): Methods, paper and evals