From cf71acdfd0ace3f8059a59727cd5bd8b9261834a Mon Sep 17 00:00:00 2001 From: Yishuo Wang Date: Mon, 16 Dec 2024 16:02:32 +0800 Subject: [PATCH] fix --- python/llm/src/ipex_llm/transformers/models/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/llm/src/ipex_llm/transformers/models/common.py b/python/llm/src/ipex_llm/transformers/models/common.py index c17fa3b59f0..4c2e830cbdd 100644 --- a/python/llm/src/ipex_llm/transformers/models/common.py +++ b/python/llm/src/ipex_llm/transformers/models/common.py @@ -181,6 +181,6 @@ def layer_norm_forward(self, hidden_states: torch.Tensor): return output.reshape(hidden_states.shape) else: return torch.nn.functional.layer_norm( - input, self.normalized_shape, + hidden_states, self.normalized_shape, self.weight, self.bias, self.eps )