From f583b56878583c1b1f15c4ea095652a9c3f83344 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Mon, 27 Oct 2025 01:06:39 +0200 Subject: [PATCH] Allow compile here --- custom_linear.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/custom_linear.py b/custom_linear.py index ff3e7fc..b65ddcc 100644 --- a/custom_linear.py +++ b/custom_linear.py @@ -101,7 +101,7 @@ class CustomLinear(nn.Linear): return torch.nn.functional.linear(input, weight, bias) - @torch.compiler.disable() + #@torch.compiler.disable() def apply_lora(self, weight): for lora_diff, lora_strength in zip(self.lora[0], self.lora[1]): if isinstance(lora_strength, list):