Update main README with loading instructions

README.md

@@ -1,5 +1,5 @@
 ---
-version: final
+version: main
 family: smollm2-1.7b
 model_name: meta-llama-Llama-3.2-1B-lr2e-05-gbs16600B
 license: mit
@@ -8,7 +8,7 @@ tags:
   - transformer
   - smollm2
 ---
-# SmolLM2 meta-llama-Llama-3.2-1B-lr2e-05-gbs16600B (Version: final)
+# SmolLM2 meta-llama-Llama-3.2-1B-lr2e-05-gbs16600B (Version: main)
 
 ## Model Details
 - **Architecture:** SmolLM2
@@ -30,3 +30,17 @@ train:
   micro_batch_size: 8
 
 ```
+
+## Model Loading and Revision System
+
+This repository hosts multiple revisions of the model.
+To load a specific revision, use the `revision` parameter. For example:
+
+```python
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+model = AutoModelForCausalLM.from_pretrained("locuslab/meta-llama-Llama-3.2-1B-lr2e-05-gbs16600B", revision="final")
+tokenizer = AutoTokenizer.from_pretrained("locuslab/meta-llama-Llama-3.2-1B-lr2e-05-gbs16600B", revision="final")
+```
+
+Replace `"final"` with the desired revision.