back

modeling_super_linear.py

@@ -546,10 +546,10 @@ class superLinear(nn.Module):
     
 "-------------------------------------------------------------------------------------------------------------------"
 class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
-    # config_class: SuperLinearConfig
+    config_class: SuperLinearConfig
 
-    def __init__(self, config):
-        super().__init__()
+    def __init__(self, config: SuperLinearConfig):
+        super().__init__(config)
   
         # the backbone keeps its own Config dataclass, so build one on‑the‑fly:
         backbone_cfg     = type("Cfg", (), config.to_dict())()