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

__all__ = [
    "DenseBackwardOLMoEConfig",
    "DenseBackwardOLMoEForCausalLM", 
    "DenseBackwardOlmoeSparseMoeBlock"
]