ur-dad-matt commited on
Commit
0cbc2ad
·
verified ·
1 Parent(s): ea019a4

fix: rename Qwen2MoE → OutlierMoE to avoid transformers 4.46+ Qwen2MoeForCausalLM namespace collision (Mac-launch-006) [modeling]

Browse files
Files changed (1) hide show
  1. 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 Qwen2MoEForCausalLM(Qwen2ForCausalLM):
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