pip install -e .[train] 可能会修改torch的版本。

[dkzyh87]

conda的虚拟环境：torch 1.13 python 3.8，执行pip install -e .[train] 换成torch 2.1.1。导致centos7系统上训练出问题，需要重新替换torch版本。

[syq163]

Thank you for your feedback. The dependency on old versions has been eliminated for both torch and transformers. Therefore, you can update and use higher versions as necessary. Currently, torch 2.1 should be suitable.