for test

intel-analytics · Oct 19, 2023 · 8a56968 · 8a56968
1 parent 3ee6879
commit 8a56968
Showing 1 changed file with 1 addition and 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
@@ -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):