From 88a60d71ab8f5603d4f57c1e86ca16b303253758 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Thu, 23 Oct 2025 23:25:20 +0300
Subject: [PATCH] Fix

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

diff --git a/custom_linear.py b/custom_linear.py
index 7154b2a..ff3e7fc 100644
--- a/custom_linear.py
+++ b/custom_linear.py
@@ -86,6 +86,8 @@ class CustomLinear(nn.Linear):
     def forward(self, input):
         if self.bias is not None:
             bias = self.bias.to(input)
+        else:
+            bias = None
         weight = self.weight.to(input)
 
         if self.scale_weight is not None: