Update modeling_Llamoe.py

modeling_Llamoe.py

@@ -251,7 +251,7 @@ def apply_rotary_pos_emb(q, k, cos, sin, position_ids=None, unsqueeze_dim=1):
 
 
 class LlamoeBlockSparseTop2MLP(nn.Module):
-    def __init__(self, config: GemmoeConfig):
+    def __init__(self, config: LlamoeConfig):
         super().__init__()
         self.ffn_dim = config.intermediate_size
         self.hidden_dim = config.hidden_size