Update _utils.py
This commit is contained in:
parent
d8c6c3e903
commit
db1c5f414a
1 changed files with 4 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue