ModuleNotFoundError: No module named 'torch._six'

[shuyhere]

System Info

wsl系统 torch=2.0.1 运行aquila-7b测试脚本时候报错

Information

• [X] The official example scripts
• [ ] My own modified scripts

Tasks

• [X] An officially supported task in the examples folder (such as T5/AltCLIP, ...)
• [ ] My own task or dataset (give details below)

Reproduction

File "/root/miniconda3/envs/flagai/lib/python3.10/site-packages/flagai/mpu/grads.py", line 23, in from torch._six import inf

更改相关文件为 from torch import inf 正常运行

Expected behavior

更改相关文件为 from torch import inf 正常运行

[BAAI-OpenPlatform]

我们在最新的1.7.1版本解决了这个问题，这个是torch的版本原因，2.0之后torch._six就没有了