diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index 61d660bce4..f49c9db1ff 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -46,6 +46,8 @@ __all__ = [
     "patch_gradient_accumulation_fix",
     "patch_compiling_bitsandbytes",
     "patch_regional_compilation",
+    "patch_layernorm",
+    "patch_torch_compile",
 ]
 
 import torch