diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index 098f5c3e47..3752d46d0a 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -58,7 +58,6 @@ __all__ = [
     "fused_linear_cross_entropy",
     "patch_unsloth_smart_gradient_checkpointing",
     "unpatch_unsloth_smart_gradient_checkpointing",
-    "create_gradient_checkpointing_buffer",
 
     "patch_compiled_autograd",
     "process_vision_info",
@@ -97,7 +96,6 @@ from unsloth_zoo.gradient_checkpointing import (
 
     patch_unsloth_smart_gradient_checkpointing,
     unpatch_unsloth_smart_gradient_checkpointing,
-    create_gradient_checkpointing_buffer,
 )
 from unsloth_zoo.loss_utils import (
     HAS_CUT_CROSS_ENTROPY,