feat(comm/attn_offload.py): support selective ckpt and cpu offload#383

Merged

sunpengsdu merged 5 commits intoInternLM:developfrom

huangting4201:feat/selective-ckpt-cpu-offload

Dec 31, 2024

Commits on Dec 3, 2024

Commits on Dec 10, 2024

feat(attn_offload.py): update attn offload manager
huangting4201
committed

Commits on Dec 17, 2024