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

熊熊的小心心

1506人浏览 · 2023-10-30 16:41:19

熊熊的小心心 · 2023-10-30 16:41:19 发布

参考：项目场景：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')

魔乐社区

魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。

更多推荐

小参数・大码力・易部署 | Qwen3.6-27B上线魔乐社区，基于昇腾的部署教程来了

继一周前模型开源发布后，千问再度开源Qwen3.6-27B —— 一个拥有270亿参数的稠密多模态模型，也是社区呼声最高的模型规格。Qwen3.6-27B 依然支持多模态思考与非思考模式，在智能体编程方面达到了旗舰级表现，全面超越前代开源旗舰 Qwen3.5-397B-A17B（总参数397B / 激活参数17B的MoE模型）。作为稠密架构，它无需MoE路由即可部署，是开发者在实用、可广泛部署规模