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