Skip to content

Optimize GPU checkpoint loading by ensuring model transfer before load_state_dict on build method #325

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

Open
nimanikoo wants to merge 1 commit into
base: main
Choose a base branch
from

feat(checkpoint-loading): ensure models are moved to GPU before loadi…

5a2ccfa
Select commit
Loading
Failed to load commit list.
Open

Optimize GPU checkpoint loading by ensuring model transfer before load_state_dict on build method #325

feat(checkpoint-loading): ensure models are moved to GPU before loadi…
5a2ccfa
Select commit
Loading
Failed to load commit list.
Microsoft GitHub Policy Service / license/cla succeeded Aug 21, 2025 in 0s

All CLA requirements met.

This check verifies that the author has agreed to a CLA with Microsoft.

View more details on Microsoft GitHub Policy Service