diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index 82bd4c979d..8f0fc30568 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -59,7 +59,7 @@ import warnings, subprocess, re, inspect, psutil, os, math
 from packaging.version import Version
 
 from unsloth_zoo.tokenizer_utils import (
-    patch_tokenizer,
+    patch_tokenizer as _patch_tokenizer,
 )
 from unsloth_zoo.patching_utils import (
     patch_compiling_bitsandbytes,
@@ -983,3 +983,11 @@ def patch_gradient_accumulation_fix(Trainer):
     exec(function, globals())
     Trainer.training_step = _unsloth_training_step
 pass
+
+
+def patch_tokenizer(model, tokenizer):
+    model, tokenizer = _patch_tokenizer(model, tokenizer)
+    if model is not None:
+        model.config.update({"unsloth_version" : __version__})
+    return model, tokenizer
+pass