transformers
4243bb84 - fix bug using FSDP V1 will lead to model device not properly set (#39177)

Commit
246 days ago
fix bug using FSDP V1 will lead to model device not properly set (#39177) * fix bug using FSDP V1 will lead to model device not properly set Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * update the code Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> --------- Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
Author
Parents
Loading