[linux] megatron转换huggingface权重报错return super().find_class(mod_name, name) No module named megatron

参考:项目场景:pytorch torch.load ModuleNotFoundError No module named ‘models‘_return super().find_class(mod_name, name) moduleno_mtl1994的博客-CSDN博客

在python脚本中,添加当前目录到path即可: 

import sys
sys.path.insert(0, '/xx/Megatron-LM')

你可能感兴趣的:(linux,linux)