ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

fix typos

Browse source
This commit is contained in:
Huang, Xinshengzi 2024-08-22 11:33:07 +08:00
parent 01ed397e7a
commit 8a5df93de2
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
python/llm/src/ipex_llm/transformers/models/baichuan.py
View file

@ -77,6 +77,7 @@ def baichuan_mlp_forward(
)) ))
return self.down_proj(self.act_fn(self.gate_proj(x)) * self.up_proj(x)) return self.down_proj(self.act_fn(self.gate_proj(x)) * self.up_proj(x))
def baichuan_model_7b_forward( def baichuan_model_7b_forward(
self, self,
input_ids: torch.LongTensor = None, input_ids: torch.LongTensor = None,