Update _utils.py

This commit is contained in:
Daniel Han 2024-11-17 16:56:26 -08:00
commit db1c5f414a

View file

@ -59,6 +59,7 @@ __all__ = [
"unpatch_unsloth_smart_gradient_checkpointing",
"patch_compiled_autograd",
"process_vision_info",
]
import torch
@ -97,6 +98,9 @@ from unsloth_zoo.loss_utils import (
HAS_CUT_CROSS_ENTROPY,
fused_linear_cross_entropy,
)
from unsloth_zoo.vision_utils import (
process_vision_info,
)
# =============================================
# Disable some warnings which can get annoying