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>