ModuleNotFoundError: No module named 'model.bert'

[xxw11]

跑example时出现了这个，是我哪里弄错了吗

[xxw11]

调整这个example在colosalai-main的位置
并改为
from model_zoo.gpt.gpt import GPTLMLoss

[FrankLeeeee]

Hi，如果你是pip install from source。应该是能直接import model_zoo的。如果不行，也可以mv model_zoo /path/to/xxx 然后export PYTHONPATH=/path/to/xxx，就可以import了

[xxw11]

测试了一下 按你说的方法确实有效 十分感谢