From 3f0e56fc2313518f8bc348bc00d10748dd4b5162 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Sun, 3 Nov 2024 15:02:38 -0800
Subject: [PATCH] Update _utils.py

---
 unsloth/models/_utils.py | 2 ++
 1 file changed, 2 insertions(+)

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