State dictionary retrieval from offloaded modules (#2619)
* added get_state_dict_from_offloaded
* cleaned
* make style
* Update src/accelerate/utils/modeling.py
Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
* implemented suggestions, refactored, make style
---------
Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
