accelerate

c0552c90 - Fix `align_module_device`, ensure only cpu tensors for `get_state_dict_offloaded_model` (#3217)

Fix `align_module_device`, ensure only cpu tensors for `get_state_dict_offloaded_model` (#3217) * only onload direct parameter descendants, move buffers to cpu, add tests * remove no longer applicable comment