PR #3204 [Utils] `align_module_device`

[Utils] `align_module_device` #3204

muellerzr merged 11 commits into huggingface:main from kylesayrs:neural-magic/align_module

implement align_module

f10afe89

add docs

4646d1a1

move to modeling utils, integrate into existing source code

6133d50f

update source, expose through utils

178b5cd9

LysandreJik requested a review from

muellerzr 1 year ago

muellerzr approved these changes on 2024-10-30

muellerzr requested a review from

BenjaminBossan 1 year ago

Suggested docstring

bab2c4d8

Rewrite for readability, add try finally

95a5a38f

Use try-finally when aligning with hook

10c6b52b

apply style

80f947ae

improve get_state_dict_from_offload readability

f908bfa3

BenjaminBossan commented on 2024-10-31

Update docstring

66e69252

rename to align_module_device, update docstring

32614628

kylesayrs changed the title ~~[Utils] `align_module`~~ [Utils] `align_module_device` 1 year ago

muellerzr merged bf4572b6 into main 1 year ago

kylesayrs deleted the neural-magic/align_module branch 1 year ago

SunMarc commented on 2024-11-04

Reviewers

muellerzr

SunMarc

BenjaminBossan

Assignees

No one assigned

Labels

None yet

Milestone

No milestone