fix typos
This commit is contained in:
parent
01ed397e7a
commit
8a5df93de2
1 changed files with 1 additions and 0 deletions
|
|
@ -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,
|
||||
|
|
|
|||
Loading…
Reference in a new issue