Assemble state dictionary for offloaded models #2156
SunMarc
merged
26 commits into
huggingface:main
from
blbadger:save-offloaded
changed meta alignment device to cpu
a345e829
reverted alignment device and init weight map
e68790c9
trace on values
61b594fe
trace on values
528fa7d0
trace on values
1d548566
Merge branch 'main' of github.com:blbadger/accelerate into peft-debug
8695c305
added offload model state dict save and test
4a33f48b
removed hook traces
635a6fc8
removed n
63365533
Update src/accelerate/accelerator.py
a8fed3bd
Update src/accelerate/utils/modeling.py
e6325b6e
Update src/accelerate/utils/modeling.py
a4c9fdfd
Update src/accelerate/utils/modeling.py
4c7516cb
suggestions and make style
fc6f627d
muellerzr
commented
on 2023-11-20
fixed circular import and make style
86ea9259
debugged test
0c40b632
Update src/accelerate/utils/modeling.py
1658d301
Update src/accelerate/utils/modeling.py
664bbc92
function level import and make style
30b656ce
muellerzr
approved these changes
on 2023-11-22
SunMarc
approved these changes
on 2023-11-22
Update src/accelerate/utils/modeling.py
041c86c9
Update tests/test_accelerator.py
cf5fa129
Update tests/test_accelerator.py
28304c58
Update src/accelerate/utils/modeling.py
e33dd912
Update src/accelerate/utils/modeling.py
0ad7230d
Update src/accelerate/utils/modeling.py
03d0fd10
make style
d1a054ce
Login to write a write a comment.
Login via GitHub