fix typos

This commit is contained in:
Huang, Xinshengzi 2024-08-22 11:33:07 +08:00
parent 01ed397e7a
commit 8a5df93de2

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))
def baichuan_model_7b_forward(
self,
input_ids: torch.LongTensor = None,