megatron转换模型格式报错 #6033
Description
[rank0]: Traceback (most recent call last):
[rank0]: File "/usr/local/lib/python3.10/dist-packages/swift/cli/export.py", line 5, in
[rank0]: export_main()
[rank0]: File "/usr/local/lib/python3.10/dist-packages/swift/llm/export/export.py", line 53, in export_main
[rank0]: return SwiftExport(args).main()
[rank0]: File "/usr/local/lib/python3.10/dist-packages/swift/llm/base.py", line 49, in main
[rank0]: result = self.run()
[rank0]: File "/usr/local/lib/python3.10/dist-packages/swift/llm/export/export.py", line 39, in run
[rank0]: from swift.megatron import convert_hf2mcore
[rank0]: File "/usr/local/lib/python3.10/dist-packages/swift/megatron/init.py", line 5, in
[rank0]: init_megatron_env()
[rank0]: File "/usr/local/lib/python3.10/dist-packages/swift/megatron/init.py", line 655, in init_megatron_env
[rank0]: _patch_megatron()
[rank0]: File "/usr/local/lib/python3.10/dist-packages/swift/megatron/init.py", line 625, in _patch_megatron
[rank0]: _patch_TransformerLayer()
[rank0]: File "/usr/local/lib/python3.10/dist-packages/swift/megatron/init.py", line 420, in _patch_TransformerLayer
[rank0]: from megatron.training import get_args
[rank0]: ModuleNotFoundError: No module named 'megatron.training'
指定了megatron-lm路径仍然报错，怎么解决