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,