From fcea6dc7d9fa92e40744a45ef8a029fc4407e6c1 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 17 Nov 2024 20:16:33 -0800 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 7243d34c3a..599f5a202f 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -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