codys12 commited on
Commit
85f2131
·
verified ·
1 Parent(s): fb00d43

Update modeling_hunyuan.py

Browse files
Files changed (1) hide show
  1. modeling_hunyuan.py +3 -3
modeling_hunyuan.py CHANGED
@@ -401,9 +401,9 @@ class HunYuanMLP(nn.Module):
401
  self.intermediate_size = config.intermediate_size * config.num_shared_expert
402
  else:
403
  self.intermediate_size = config.intermediate_size
404
- self.gate_proj = nn.Linear(self.hidden_size, self.intermediate_size, bias=False, dtype=None)
405
- self.up_proj = nn.Linear(self.hidden_size, self.intermediate_size, bias=False, dtype=None)
406
- self.down_proj = nn.Linear(self.intermediate_size, self.hidden_size, bias=False, dtype=None)
407
  self.act_fn = ACT2FN[config.hidden_act]
408
 
409
  def forward(self, x):
 
401
  self.intermediate_size = config.intermediate_size * config.num_shared_expert
402
  else:
403
  self.intermediate_size = config.intermediate_size
404
+ self.gate_proj = nn.Linear(self.hidden_size, self.intermediate_size, bias=False)
405
+ self.up_proj = nn.Linear(self.hidden_size, self.intermediate_size, bias=False)
406
+ self.down_proj = nn.Linear(self.intermediate_size, self.hidden_size, bias=False)
407
  self.act_fn = ACT2FN[config.hidden_act]
408
 
409
  def forward(self, x):