Update modeling_custom_minilm.py

modeling_custom_minilm.py

@@ -109,7 +109,7 @@ class SpaceTimeSelfAttention(nn.Module):
 # 2) Full Space–Time–MiniLM Model
 # ----------------------------
 class SpaceTimeMiniLM(PreTrainedModel):
-    config_class = AutoConfig
+    config_class = SpaceTimeMiniLMConfig
     def __init__(self, config):
         super().__init__(config)
         self.base = AutoModel.from_config(config)