From cc7f1b99a38b0b7e14322dde0eff2ad18b38c304 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 4 Jan 2025 00:42:39 -0800 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 2 -- 1 file changed, 2 deletions(-) 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,