HuggingFaceTB
/

SmolLM2-360M-intermediate-checkpoints

Inference Endpoints

Model card Files Files and versions Community

eliebak HF staff commited on 1 day ago

Commit

b5cf95e

·

verified ·

1 Parent(s): ec0c8d3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -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-360M-intermediate-checkpoints"
-revision = "step-320000" # 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)

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 checkpoint = "HuggingFaceTB/SmolLM2-360M-intermediate-checkpoints"
+revision = "step-160000" # 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)