File size: 479 Bytes
d606666 94724ad aaaf523 94724ad d606666 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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"
] |