diff --git a/mergekit/scripts/tokensurgeon.py b/mergekit/scripts/tokensurgeon.py index 4f9b226f..60ee05dd 100644 --- 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: