Debugging

config.json

@@ -3,9 +3,9 @@
     "SharedSpaceDecoderForCausalLM"
   ],
   "auto_map": {
-    "AutoConfig": "models.shared_space_config:SharedSpaceDecoderConfig",
-    "AutoModel": "models.shared_space_decoder:SharedSpaceDecoderModel",
-    "AutoModelForCausalLM": "layers.task_heads:SharedSpaceDecoderForCausalLM"
+    "AutoConfig": "configuration_shared_space_decoder.SharedSpaceDecoderConfig",
+    "AutoModel": "modeling_shared_space_decoder.SharedSpaceDecoderModel",
+    "AutoModelForCausalLM": "modeling_shared_space_decoder.SharedSpaceDecoderForCausalLM"
   },
   "attention_backend": "flash_attention_2",
   "attention_bias": false,