File size: 312 Bytes
52bb403 |
1 2 3 4 5 6 7 8 9 10 |
# 导出自定义配置和模型类
from .configuration_custom import DenseBackwardOLMoEConfig
from .modeling_custom import DenseBackwardOLMoEForCausalLM, DenseBackwardOlmoeSparseMoeBlock
__all__ = [
"DenseBackwardOLMoEConfig",
"DenseBackwardOLMoEForCausalLM",
"DenseBackwardOlmoeSparseMoeBlock"
]
|