使用Qwen3模型进行训练,在初始化阶段会有两次权重加载过程:
首次只做meta初始化获取到model,分片后再加载权重
Python 3.11.10 torch 2.6.0 transformers 5.13.0
Thanks for contributing 🎉!
Checklist
🐛 Describe the bug
使用Qwen3模型进行训练,在初始化阶段会有两次权重加载过程:
Expected behavior
首次只做meta初始化获取到model,分片后再加载权重
Additional context
Environment info
Python 3.11.10
torch 2.6.0
transformers 5.13.0
Thanks for contributing 🎉!