Update modeling_hunyuan.py
Browse files- 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
|
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):
|
|
|
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):
|