| 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" | |
| ] |