From b1d9a2bd72995e259cfab66b414abddac41cc873 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 4 Nov 2024 01:23:24 -0800 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 7a8dcedada..182fdb1dd4 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -75,6 +75,11 @@ from unsloth_zoo.patching_utils import ( from unsloth_zoo.gradient_checkpointing import ( Unsloth_Offloaded_Gradient_Checkpointer, unsloth_offloaded_gradient_checkpoint, + patch_unsloth_gradient_checkpointing, + unpatch_unsloth_gradient_checkpointing, + + Unsloth_Gradient_Checkpointer, + unsloth_gradient_checkpoint, patch_gradient_checkpointing, unpatch_gradient_checkpointing, )