Update README.md

README.md

@@ -38,10 +38,16 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 ```
 
 ## Using specific checkpoints
+To make experimentation easier, we uploaded multiple checkpoints from each pretraining and midtraining stage as tagged releases. These checkpoints follow a consistent naming convention.
+
+Checkpoints from different stages use the stage name as a prefix: `base`, `mid_1`, `mid_2`, `mid_3`, and `mid_4`. Final checkpoints are marked with the final suffix, while intermediate checkpoints are identified by their checkpoint numbers.
+
+For example, the final checkpoint from midtraining stage 4 is named `mid_4_final`.
+
 To use a specific checkpoint, use the `revision` argument:
 
 ```
-model = AutoModelForCausalLM.from_pretrained("llm360/k2-v2-instruct", device_map="auto", revision="base_0720000")
+model = AutoModelForCausalLM.from_pretrained("llm360/k2-v2", device_map="auto", revision="base_0720000")
 ```
 ---