SequentialLearning
/

SuperLinear

mixture-of-experts

Model card Files Files and versions

razmars commited on May 13

Commit

cbe778a

·

1 Parent(s): 298a933

fix git

Files changed (1) hide show

modeling_super_linear.py +1 -1

modeling_super_linear.py CHANGED Viewed

@@ -546,7 +546,7 @@ class superLinear(nn.Module):
 "-------------------------------------------------------------------------------------------------------------------"
 class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
-    # config_class: SuperLinearConfig
     def __init__(self, config):
         super().__init__(config)

 "-------------------------------------------------------------------------------------------------------------------"
 class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
+    config_class: SuperLinearConfig
     def __init__(self, config):
         super().__init__(config)