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

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged

sunpengsdu merged 5 commits into InternLM:develop from huangting4201:feat/selective-ckpt-cpu-offload

Dec 31, 2024

Merged

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

fix(conflicts): fix conflicts from merging develop

Annotations

2 warnings

The logs for this run have expired and are no longer available.