Update __init__.py
Browse files- __init__.py +3 -3
__init__.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
# 导出自定义配置和模型类
|
2 |
-
from .
|
3 |
-
from .
|
4 |
|
5 |
__all__ = [
|
6 |
"DenseBackwardOLMoEConfig",
|
7 |
"DenseBackwardOLMoEForCausalLM",
|
8 |
"DenseBackwardOlmoeSparseMoeBlock"
|
9 |
-
]
|
|
|
1 |
# 导出自定义配置和模型类
|
2 |
+
from .configuration_densebackward_olmoe import DenseBackwardOLMoEConfig
|
3 |
+
from .modeling_densebackward_olmoe import DenseBackwardOLMoEForCausalLM, DenseBackwardOlmoeSparseMoeBlock
|
4 |
|
5 |
__all__ = [
|
6 |
"DenseBackwardOLMoEConfig",
|
7 |
"DenseBackwardOLMoEForCausalLM",
|
8 |
"DenseBackwardOlmoeSparseMoeBlock"
|
9 |
+
]
|