Fix tokensurgeon failing to save when optional tensors are present

cg123 · cg123 · commit ce158c7cae96 · 2025-06-10T20:35:25.000-07:00
diff --git a/mergekit/scripts/tokensurgeon.py b/mergekit/scripts/tokensurgeon.py
@@ -738,7 +738,7 @@ def main(
             tensor = new_lm_head
         else:
             tensor = cache.get(options.model).get_tensor(
-                weight_info.name, aliases=weight_info.aliases
+                weight_info.name, aliases=weight_info.aliases, raise_on_missing=False
             )
         if tensor is None:
             if weight_info.optional:

Original file line number	Diff line number	Diff line change
`@@ -738,7 +738,7 @@ def main(`
`738`	`738`	`tensor = new_lm_head`
`739`	`739`	`else:`
`740`	`740`	`tensor = cache.get(options.model).get_tensor(`
`741`		`- weight_info.name, aliases=weight_info.aliases`
	`741`	`+ weight_info.name, aliases=weight_info.aliases, raise_on_missing=False`
`742`	`742`	`)`
`743`	`743`	`if tensor is None:`
`744`	`744`	`if weight_info.optional:`