Accelerate Utilities Follow-up

[kylesayrs]

Purpose

• Add more utilities not covered by Accelerate Utilities #193
• Add tests for offloading functions
• Reduce code complexity by using utility functions
• Fix bug in update_offload_data and expand capabilities

Changes

• Implement align_module_device, which has already been upstreamed and is just copied here until we bump the required accelerate version
• Implement disable_hf_hook and utilize in initialize_module_for_quantization
  • Note that the previous "dictionary recreation" logic is no longer needed. It was previously necessary since the newly attached parameters were not updated in the state dict, however this is now done by register_offload_parameter
• Fix bug in update_offload_data where data would not be updated properly if the module was offloaded and weights_map dataset was being improperly gotten
• Cover raw dictionary case in update_offload_data

Testing

• Added tests in tests/test_utils/test_offload.py
• Added offloaded test in tests/test_quantization/lifecycle/test_initialize.py