Use instruction tuned checkpoints for SmoLLM verification.

haozha111 · copybara-github · commit c5ab8923d4bb · 2025-05-15T17:56:42.000-07:00
PiperOrigin-RevId: 759377620
diff --git a/ai_edge_torch/generative/examples/smollm/verify.py b/ai_edge_torch/generative/examples/smollm/verify.py
@@ -43,8 +43,8 @@
     "The version of SmolLm to verify.",
 )
 _CHECKPOINT = {
-    "v1": "HuggingFaceTB/SmolLM-135M",
-    "v2": "HuggingFaceTB/SmolLM2-135M",
+    "v1": "HuggingFaceTB/SmolLM-135M-Instruct",
+    "v2": "HuggingFaceTB/SmolLM2-135M-Instruct",
 }
 
 _BUILDER = {

Original file line number	Diff line number	Diff line change
`@@ -43,8 +43,8 @@`
`43`	`43`	`"The version of SmolLm to verify.",`
`44`	`44`	`)`
`45`	`45`	`_CHECKPOINT = {`
`46`		`- "v1": "HuggingFaceTB/SmolLM-135M",`
`47`		`- "v2": "HuggingFaceTB/SmolLM2-135M",`
	`46`	`+ "v1": "HuggingFaceTB/SmolLM-135M-Instruct",`
	`47`	`+ "v2": "HuggingFaceTB/SmolLM2-135M-Instruct",`
`48`	`48`	`}`
`49`	`49`
`50`	`50`	`_BUILDER = {`