autoprogrammer commited on
Commit
d606666
·
verified ·
1 Parent(s): 965912e

Update __init__.py

Browse files
Files changed (1) hide show
  1. __init__.py +3 -3
__init__.py CHANGED
@@ -1,9 +1,9 @@
1
  # 导出自定义配置和模型类
2
- from .configuration_custom import DenseBackwardOLMoEConfig
3
- from .modeling_custom import DenseBackwardOLMoEForCausalLM, DenseBackwardOlmoeSparseMoeBlock
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
+ ]