用这个damo/nlp_gpt3_text-generation_chinese-base 模型本地跑的时候报这个包没有,麻烦问下这个怎么解决呢?
您好,torch2.0 中删除了 torch._six,一个比较直接的解决方案是将 /Users/admin/opt/anaconda3/envs/gpt/lib/python3.8/site-packages/megatron_util/mpu/grads.py 第 19 行的 from torch._six import inf 替换为 from torch import inf,后续我们会对此进行兼容性修复~,此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”