stupid

modeling_super_linear.py

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