From 8a5df93de2b82319799aca3443015b04e0fc9a55 Mon Sep 17 00:00:00 2001 From: "Huang, Xinshengzi" Date: Thu, 22 Aug 2024 11:33:07 +0800 Subject: [PATCH] fix typos --- python/llm/src/ipex_llm/transformers/models/baichuan.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/llm/src/ipex_llm/transformers/models/baichuan.py b/python/llm/src/ipex_llm/transformers/models/baichuan.py index 1f39c0bd..d454a97c 100644 --- a/python/llm/src/ipex_llm/transformers/models/baichuan.py +++ b/python/llm/src/ipex_llm/transformers/models/baichuan.py @@ -77,6 +77,7 @@ def baichuan_mlp_forward( )) return self.down_proj(self.act_fn(self.gate_proj(x)) * self.up_proj(x)) + def baichuan_model_7b_forward( self, input_ids: torch.LongTensor = None,