From 8a569686ba3132232666e377ebdfd032df470232 Mon Sep 17 00:00:00 2001 From: WeiguangHan Date: Thu, 19 Oct 2023 09:59:56 +0800 Subject: [PATCH] for test --- python/llm/src/bigdl/llm/transformers/low_bit_linear.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/llm/src/bigdl/llm/transformers/low_bit_linear.py b/python/llm/src/bigdl/llm/transformers/low_bit_linear.py index 4940b13eef7b..1f32079511a8 100644 --- a/python/llm/src/bigdl/llm/transformers/low_bit_linear.py +++ b/python/llm/src/bigdl/llm/transformers/low_bit_linear.py @@ -385,7 +385,7 @@ def forward(self, x: torch.Tensor): result = result.view(new_shape) if self.bias is not None: result += self.bias - return result + return result.to(x.dtype) class FP16Linear(nn.Linear):