Update _utils.py

This commit is contained in:
Daniel Han 2025-01-04 00:42:39 -08:00
commit cc7f1b99a3

View file

@ -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,