Fsdp 2 - #3

Open

andreaskoepf wants to merge 10 commits into

mainfrom

Commits on Jan 24, 2025

add fsdp2
samsja
committed
9015634
add fsdp2
samsja
committed
c4f70bc
wip
samsja
committed
b6044f5
fix ckpt
samsja
committed
36f954a
fix ckpt
samsja
committed
3d387d4
update param
samsja
committed
eb1b796
remove llama decoder import
samsja
committed
833ba4a
add back forward
samsja
committed
d104a56
Merge pull request #1 from samsja/fsdp_2
andreaskoepf
authored
ef14fcb
black formatting
andreaskoepf
committed
e681b87