[6/N] MoE Refactor: Cleanup MoE-related configs (#8849)
This commit is contained in:
@@ -135,7 +135,6 @@ class Grok1MoE(nn.Module):
|
||||
intermediate_size=intermediate_size,
|
||||
params_dtype=params_dtype,
|
||||
quant_config=quant_config,
|
||||
tp_size=tp_size,
|
||||
activation="gelu",
|
||||
**kwargs,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user