已开启
[Bug]: 初始化阶段会有两次权重加载过程 #365
pangxz创建于  20 天前
pangxz
pangxz成员
20 天前 创建

Checklist

🐛 Describe the bug

使用Qwen3模型进行训练,在初始化阶段会有两次权重加载过程:

  1. 第一次是在meta初始化阶段from_pretrained会触发读权重到cpu,大概耗时2~5min,有进度条提醒
  2. 第二次是分片后实际初始化阶段,耗时反而更久10min左右,且没加载进度打印

Expected behavior

首次只做meta初始化获取到model,分片后再加载权重

Additional context

Environment info

Python 3.11.10
torch 2.6.0
transformers 5.13.0

Thanks for contributing 🎉!

likedislike
pangxzpangxz成员
20 天前 添加了label:bug