Update __init__.py

__init__.py

@@ -1,6 +1,10 @@
 # 导出自定义配置和模型类
-from .configuration_custom import DenseBackwardOLMoEConfig
-from .modeling_custom import DenseBackwardOLMoEForCausalLM, DenseBackwardOlmoeSparseMoeBlock
+from .configuration_densebackward_olmoe0125 import DenseBackwardOLMoEConfig
+from .modeling_densebackward_olmoe0125 import DenseBackwardOLMoEForCausalLM, DenseBackwardOlmoeSparseMoeBlock
+
+# 显式注册模型类型
+from transformers.models.auto.configuration_auto import CONFIG_MAPPING
+from transformers.models.auto.modeling_auto import MODEL_FOR_CAUSAL_LM_MAPPING
 
 __all__ = [
     "DenseBackwardOLMoEConfig",