fix

intel · Jan 22, 2025 · 5ea9a8b · 5ea9a8b
1 parent f662366
commit 5ea9a8b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/llm/src/ipex_llm/transformers/low_bit_linear.py b/python/llm/src/ipex_llm/transformers/low_bit_linear.py
@@ -659,7 +659,9 @@ def forward(self, x: torch.Tensor):
             if x_2d.size(0) == 1:
                 import xe_linear
                 result = xe_linear.linear_forward_vec(x_2d, self.weight.data,
-                    self.in_len, self.out_len, self.qtype)
+                                                      self.in_len, self.out_len, self.qtype)
+                if self.bias is not None:
+                    result += self.bias
                 return result.view(new_shape)
 
             if len(x_shape) == 3: