__init__.py

# 导出自定义配置和模型类
from .configuration_densebackward_olmoe0125_v1 import DenseBackwardOLMoEConfig
from .modeling_densebackward_olmoe0125_v1 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",
    "DenseBackwardOLMoEForCausalLM", 
    "DenseBackwardOlmoeSparseMoeBlock"
]