Update README.md

README.md

@@ -45,6 +45,40 @@ outputs = model.generate(**input_ids, max_new_tokens=100)
 print(tokenizer.decode(outputs[0]))
 ```
 
+To load a different checkpoint use, e.g., for iteration 2500,
+
+```python
+model = AutoModelForCausalLM.from_pretrained("Zyphra/Zamba-7B-v1-phase1", device_map="auto", torch_dtype=torch.bfloat16, revision="iter2500")
+```
+
+The default iteration is the fully trained phase 1 model, corresponding to iteration 462070. This is the number of iterations performed by training the model starting from random initialization. See [arXiv:2405.16712](https://arxiv.org/abs/2405.16712) for more details on training.
+
+## Model Details
+
+Zamba utilizes a unique hybrid SSM architecture. This architecture consists of a backbone of Mamba layers interspersed with a shared attention layer. This attention has shared weights to minimize the parameter cost of the model. We find that concatenating the original model embeddings to the input to this attention block improves performance, likely due to better maintenance of information across depth. 
+
+
+<center>
+<img src="https://cdn-uploads.huggingface.co/production/uploads/65c05e75c084467acab2f84a/IGK562oVTFSOQbpLavu7E.png" width="300" alt="Zamba architecture">
+</center>
+
+
+## Performance
+
+We find that Zamba performs significantly better than existing open models (with open datasets and training details) at this scale. However, it performs slightly worse than the leading open-weight models at the 7B scale. Most of this difference derives from MMLU and reasoning evaluations. Zamba, however, is trained on significantly fewer tokens than these models and is the most sample efficient model in terms of performance per training tokens.
+
+
+<center>
+<img src="https://cdn-uploads.huggingface.co/production/uploads/64e40335c0edca443ef8af3e/HUujLyiYpPgwz_fpdw0iG.png" width="350" alt="Zamba performance">
+</center>
+
+
+Due to its SSM architecture, Zamba is extremely efficient in inference, substantially outperforming comparable 7B and 8B models in inference latency as well as memory cost of generation due to its substantially diminished KV cache.
+
+<center>
+<img src="https://cdn-uploads.huggingface.co/production/uploads/65c05e75c084467acab2f84a/cghYPnDbdzweT1b2RyiXA.png" width="400" alt="Zamba performance">
+</center>
+
 ## Citation
 
 If you find Zamba useful in your work please cite it as:
@@ -60,4 +94,4 @@ If you find Zamba useful in your work please cite it as:
 
 ## Notice
 
-Zamba is a pretrained base model and therefore does not have any moderation mechanism.
+Zamba is a pretrained base model and therefore does not have any moderation mechanism. In addition, one should not expect good chat performance, as this model was not fine-tuned for chat.