fix: rename Qwen2MoE → OutlierMoE to avoid transformers 4.46+ Qwen2MoeForCausalLM namespace collision (Mac-launch-006) [modeling]
Browse files- modeling_outlier_moe.py +1 -1
modeling_outlier_moe.py
CHANGED
|
@@ -72,7 +72,7 @@ class Qwen2MoEMLP(nn.Module):
|
|
| 72 |
return (shared_out + expert_out).reshape(original_shape)
|
| 73 |
|
| 74 |
|
| 75 |
-
class
|
| 76 |
def __init__(self, config):
|
| 77 |
super().__init__(config)
|
| 78 |
|
|
|
|
| 72 |
return (shared_out + expert_out).reshape(original_shape)
|
| 73 |
|
| 74 |
|
| 75 |
+
class OutlierMoEForCausalLM(Qwen2ForCausalLM):
|
| 76 |
def __init__(self, config):
|
| 77 |
super().__init__(config)
|
| 78 |
|