Code

[Utils] align_module_device#3204

Code

Merged

muellerzr merged 11 commits intohuggingface:mainhuggingface/accelerate:mainfrom kylesayrs:neural-magic/align_modulekylesayrs/accelerate:neural-magic/align_moduleCopy head branch name to clipboardNov 1, 2024

+105-32

Commits

Commits on Oct 29, 2024

• implement align_module

  kylesayrs

  committed
  f10afe8

  Copy full SHA for f10afe8

• add docs

  kylesayrs

  committed
  4646d1a

  Copy full SHA for 4646d1a

• move to modeling utils, integrate into existing source code

  kylesayrs

  committed
  6133d50

  Copy full SHA for 6133d50

• update source, expose through utils

  kylesayrs

  committed
  178b5cd

  Copy full SHA for 178b5cd

Commits on Oct 30, 2024

• Suggested docstring

  Show description for bab2c4d
  

  kylesayrs

  and

  muellerzr

  authored
  bab2c4d

  Copy full SHA for bab2c4d

• Rewrite for readability, add try finally

  Show description for 95a5a38
  

  kylesayrs

  and

  muellerzr

  authored
  95a5a38

  Copy full SHA for 95a5a38

• Use try-finally when aligning with hook

  Show description for 10c6b52
  

  kylesayrs

  and

  muellerzr

  authored
  10c6b52

  Copy full SHA for 10c6b52

• apply style

  kylesayrs

  committed
  80f947a

  Copy full SHA for 80f947a

• improve get_state_dict_from_offload readability

  kylesayrs

  committed
  f908bfa

  Copy full SHA for f908bfa

Commits on Oct 31, 2024

• Update docstring

  Show description for 66e6925
  

  kylesayrs

  and

  BenjaminBossan

  authored
  66e6925

  Copy full SHA for 66e6925

• rename to align_module_device, update docstring

  kylesayrs

  committed
  3261462

  Copy full SHA for 3261462