BERT示例运行错误，ColossalAI-Examples/language/bert/sequene_parallel/

[ZXM1063694570]

🐛 Describe the bug

使用了最新提供的Dockerhub上的镜像0.1.8，但是在运行BERT序列并行案例：ColossalAI-Examples/language/bert/sequene_parallel/时候仍不能正常运行，提示缺少相关包：
Traceback (most recent call last):
File "/workspace/ColossalAI-Examples/language/bert/sequene_parallel/train.py", line 10, in
from model.bert import BertForPretrain
File "/workspace/ColossalAI-Examples/language/bert/sequene_parallel/model/bert.py", line 12, in
from colossalai.builder.pipeline import partition_uniform
ModuleNotFoundError: No module named 'colossalai.builder.pipeline'

Environment

docker镜像：docker pull hpcaitech/colossalai:0.1.8

[tianboh]

I have met this error as well. Use from colossalai.pipeline.utils import partition_uniform instead.

[ZXM1063694570]

我也遇到过这个错误。改为使用from colossalai.pipeline.utils import partition_uniform。

It worked. Thank you