modeling

modeling_super_linear.py

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