Update main branch model card with loading instructions.

README.md

@@ -1,5 +1,5 @@
 ---
-version: final
+version: main
 family: smollm2-360m
 model_name: smollm2-360m-all_raw_folders_baseline-300B-mbs16-gbs1024-16feb-lr2e-05-gbs16
 license: mit
@@ -8,7 +8,7 @@ tags:
   - transformer
   - smollm2
 ---
-# SmolLM2 smollm2-360m-all_raw_folders_baseline-300B-mbs16-gbs1024-16feb-lr2e-05-gbs16 (Version: final)
+# SmolLM2 smollm2-360m-all_raw_folders_baseline-300B-mbs16-gbs1024-16feb-lr2e-05-gbs16 (Version: main)
 
 ## Model Details
 - **Architecture:** SmolLM2
@@ -44,3 +44,18 @@ train:
   tie_embeddings: false
 
 ```
+
+
+## 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/mix_ift_v3-smollm2-360m-smollm2-360m-all_raw_folders_baseline", revision="final")
+tokenizer = AutoTokenizer.from_pretrained("locuslab/mix_ift_v3-smollm2-360m-smollm2-360m-all_raw_folders_baseline", revision="final")
+```
+
+Replace `"final"` with the desired revision.