| # 导出自定义配置和模型类 | |
| 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" | |
| ] | |