Update custom_linear.py

2025-10-28 01:55:25 +02:00
parent 90908df260
commit 2633119505
1 changed files with 1 additions and 1 deletions
@@ -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):