transformers
4f0bf986 - Fix `save_pretrained` for partially offloaded models (#34890)

Commit
1 year ago
Fix `save_pretrained` for partially offloaded models (#34890) * delete unnecessary reference Signed-off-by: Kyle Sayers <kylesayrs@gmail.com> * update comment, explicit delete state_dict * Update src/transformers/modeling_utils.py Co-authored-by: Zach Mueller <muellerzr@gmail.com> * fix style Signed-off-by: Kyle Sayers <kylesayrs@gmail.com> --------- Signed-off-by: Kyle Sayers <kylesayrs@gmail.com> Co-authored-by: Zach Mueller <muellerzr@gmail.com>
Author
Parents
Loading