File size: 530 Bytes
52bb403 59e22b7 e562a26 52bb403 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
# 导出自定义配置和模型类
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"
]
|