Update model files

config.json

@@ -2,10 +2,10 @@
   "architectures": [
     "SharedSpaceDecoderForCausalLM"
   ],
+  "model_type": "shared_subspace_decoder",
   "auto_map": {
-    "AutoConfig": "configuration_shared_subspace_decoder.SharedSpaceDecoderConfig",
-    "AutoModel": "modeling_shared_subspace_decoder.SharedSpaceDecoderModel",
-    "AutoModelForCausalLM": "modeling_shared_subspace_decoder.SharedSpaceDecoderForCausalLM"
+    "AutoConfig": "shared_space_config.SharedSpaceDecoderConfig",
+    "AutoModelForCausalLM": "task_heads.SharedSpaceDecoderForCausalLM"
   },
   "attention_backend": "flash_attention_2",
   "attention_bias": false,
@@ -23,7 +23,6 @@
   "kv_shared_dim": 96,
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 1024,
-  "model_type": "shared_subspace_decoder",
   "nope_dims": 32,
   "norm_type": "rmsnorm",
   "num_attention_heads": 12,
@@ -45,4 +44,4 @@
   "vocab_rank": null,
   "vocab_size": 50257,
   "vocab_subspace": false
-}
+}