Update README.md

README.md

@@ -36,14 +36,23 @@ The newest quants uploaded here need at least commit f4ab2a41476600a98067a9474ea
 
 # Original Model Card
 
+<a href="https://www.gradient.ai" target="_blank"><img src="https://cdn-uploads.huggingface.co/production/uploads/655bb613e8a8971e89944f3e/TSa3V8YpoVagnTYgxiLaO.png" width="200"/></a>
+
 # Llama-3 8B Gradient Instruct 1048k
+
+Join our custom agent and long context (262k-1M+) waitlist: https://forms.gle/L6TDY7dozx8TuoUv7
+
 Gradient incorporates your data to deploy autonomous assistants that power critical operations across your business. If you're looking to build custom AI models or agents, email us a message [email protected].
 
 For more info see our [End-to-end development service for custom LLMs and AI systems](https://gradient.ai/development-lab)
 
+[Join our Discord](https://discord.com/invite/2QVy2qt2mf)
+
 This model extends LLama-3 8B's context length from 8k to > 1040K, developed by Gradient, sponsored by compute from [Crusoe Energy](https://huggingface.co/crusoeai). It demonstrates that SOTA LLMs can learn to operate on long context with minimal training by appropriately adjusting RoPE theta. We trained on 830M tokens for this stage, and 1.4B tokens total for all stages, which is < 0.01% of Llama-3's original pre-training data.
 
-![image/png](https://cdn-uploads.huggingface.co/production/uploads/6585dc9be92bc5f258156bd6/6MKLoX2ruLIaREiyb6coO.png)
+**Update (5/3): We further fine-tuned our model to strengthen its assistant-like chat ability as well. The NIAH result is updated.** 
+
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/6585dc9be92bc5f258156bd6/-qaI__83ksClzoJzlqZjq.png)
 
 **Approach:**
 
@@ -59,7 +68,7 @@ Notably, we layered parallelism on top of Ring Attention with a custom network t
 
 **Data:**
 
-For training data, we generate long contexts by augmenting [SlimPajama](https://huggingface.co/datasets/cerebras/SlimPajama-627B).
+For training data, we generate long contexts by augmenting [SlimPajama](https://huggingface.co/datasets/cerebras/SlimPajama-627B). We also fine-tune on a chat dataset based on UltraChat [4], following a similar recipe for data augmentation to [2].
 
 **Progressive Training Details:**
 
@@ -68,7 +77,7 @@ For training data, we generate long contexts by augmenting [SlimPajama](https://
 | Initialize From        | LLaMA-3 8B| 65K       | 262K      | 524k      |
 | Sequence Length 2^N    | 16        | 18        | 19        | 20        |
 | RoPE theta             | 15.3 M    | 207.1 M   | 1.06B     | 2.80B     |
-| Batch Size             | 1         | 1         | 16         | 16         |
+| Batch Size             | 1         | 1         | 16         | 8         |
 | Gradient Accumulation Steps | 32  | 16        | 1         | 1         |
 | Steps                  | 30        | 24        | 50        | 50        |
 | Total Tokens           | 62914560  | 100663296 | 419430400 | 838860800 |
@@ -77,9 +86,37 @@ For training data, we generate long contexts by augmenting [SlimPajama](https://
 | GPU Type               | NVIDIA L40S | NVIDIA L40S | NVIDIA L40S | NVIDIA L40S |
 | Minutes to Train (Wall)| 202       | 555       | 61        | 87        |
 
-**Quants**:
-- [GGUF](https://huggingface.co/crusoeai/Llama-3-8B-Instruct-1048k-GGUF)
+
+**Evaluation:**
+
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/6585dc9be92bc5f258156bd6/mWxIGZNi3ejlmeIDWafKu.png)
+
+```
+EVAL_MAX_CONTEXT_LENGTH=1040200
+EVAL_MIN_CONTEXT_LENGTH=100
+EVAL_CONTEXT_INTERVAL=86675
+EVAL_DEPTH_INTERVAL=0.2
+EVAL_RND_NUMBER_DIGITS=8
+
+HAYSTACK1:
+    EVAL_GENERATOR_TOKENS=25
+
+HAYSTACK2:
+    EVAL_CONTEXT_INTERVAL=173350
+    EVAL_GENERATOR_TOKENS=150000
+
+HAYSTACK3:
+    EVAL_GENERATOR_TOKENS=925000
+```
+
+All boxes not pictured for Haystack 1 and 3 are 100% accurate. Haystacks 1,2 and 3 are further detailed in this [blog post](https://gradient.ai/blog/the-haystack-matters-for-niah-evals).
+
+**Quants:**
+- [GGUF by Crusoe](https://huggingface.co/crusoeai/Llama-3-8B-Instruct-1048k-GGUF). Note that you need to add 128009 as [special token with llama.cpp](https://huggingface.co/gradientai/Llama-3-8B-Instruct-262k/discussions/13).
 - [MLX-4bit](https://huggingface.co/mlx-community/Llama-3-8B-Instruct-1048k-4bit)
+- [Ollama](https://ollama.com/library/llama3-gradient)
+- vLLM docker image, recommended to load via `--max-model-len 32768`
+- If you are interested in a hosted version, drop us a mail below.
 
 ## The Gradient AI Team
 
@@ -99,6 +136,9 @@ Drop an email to [[email protected]](mailto:[email protected])
 
 [3] https://github.com/jzhang38/EasyContext
 
+[4] Ning Ding, Yulin Chen, Bokai Xu, Yujia Qin, Zhi Zheng, Shengding Hu, Zhiyuan
+Liu, Maosong Sun, and Bowen Zhou. Enhancing chat language models by scaling
+high-quality instructional conversations. arXiv preprint arXiv:2305.14233, 2023.
 
 ----