ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update _utils.py

Browse source
This commit is contained in:
Daniel Han 2024-11-17 20:16:33 -08:00
commit fcea6dc7d9
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/models/_utils.py
View file

@ -61,6 +61,7 @@ __all__ = [
"patch_compiled_autograd",
"process_vision_info",
"unsloth_compile_transformers",
]
import torch
@ -103,6 +104,9 @@ from unsloth_zoo.loss_utils import (
from unsloth_zoo.vision_utils import (
process_vision_info,
)
from unsloth_zoo.compiler import (
unsloth_compile_transformers,
)
# =============================================
# Disable some warnings which can get annoying