From 0cba1edd4eeab8bc8b4d0295cfbcf3f33909ec53 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sun, 30 Nov 2025 17:53:28 +0200
Subject: [PATCH] Better just not compile this as it's causing issues

---
 custom_linear.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/custom_linear.py b/custom_linear.py
index e6cd2bf..5b047b1 100644
--- a/custom_linear.py
+++ b/custom_linear.py
@@ -96,6 +96,7 @@ class CustomLinear(nn.Linear):
 
         if not allow_compile:
             self._get_weight_with_lora = torch.compiler.disable()(self._get_weight_with_lora)
+            self.forward = torch.compiler.disable()(self.forward)
     
     def set_lora_diffs(self, lora_diffs, device=torch.device("cpu")):
         self.lora_diffs = []