Update README.md

README.md

@@ -15,7 +15,7 @@ We are releasing an intermediate checkpoint of SmolLM2 to enable further researc
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 checkpoint = "HuggingFaceTB/SmolLM2-135M-intermediate-checkpoints"
-revision = "step-1680000" # replace by the revision you want
+revision = "step-240000" # replace by the revision you want
 device = torch.device("cuda" if torch.cuda.is_available() else "mps" if hasattr(torch, 'mps') and torch.mps.is_available() else "cpu")
 tokenizer = AutoTokenizer.from_pretrained(checkpoint, revision=revision)
 model = AutoModelForCausalLM.from_pretrained(checkpoint, revision=revision).to(device)
@@ -26,7 +26,7 @@ print(tokenizer.decode(outputs[0]))
 
 ## Training Details
 For comprehensive information about SmolLM2 training methodology, please refer to:
-- Our [model page](https://huggingface.co/path/to/smollm2)
+- Our [model page](https://huggingface.co/HuggingFaceTB/SmolLM2-135M)
 - Our [GitHub repository](https://github.com/huggingface/smollm)
 - Our [paper](https://huggingface.co/papers/2502.02737)