Update __init__.py

__init__.py

@@ -1,7 +1,14 @@
-# 导出自定义配置和模型类
 from .configuration_densebackward_olmoe import DenseBackwardOLMoEConfig
 from .modeling_densebackward_olmoe import DenseBackwardOLMoEForCausalLM, DenseBackwardOlmoeSparseMoeBlock
 
+# 显式注册模型类型
+from transformers.models.auto.configuration_auto import CONFIG_MAPPING
+from transformers.models.auto.modeling_auto import MODEL_FOR_CAUSAL_LM_MAPPING
+
+# 注册配置和模型类
+CONFIG_MAPPING.register("DenseBackward_olmoe", DenseBackwardOLMoEConfig)
+MODEL_FOR_CAUSAL_LM_MAPPING.register((DenseBackwardOLMoEConfig,), DenseBackwardOLMoEForCausalLM)
+
 __all__ = [
     "DenseBackwardOLMoEConfig",
     "DenseBackwardOLMoEForCausalLM",