Skip to content

Conversation

kausv
Copy link
Contributor

@kausv kausv commented Aug 26, 2025

Summary:
We were skipping loading of state dict for CPU models causing the state dict comparison between sharded and unsharded model to not match.
I removed this condition so that we skip only for meta devices.

https://www.internalfb.com/intern/test/562950052233328

Differential Revision: D81039244

Summary:
We were skipping loading of state dict for CPU models causing the state dict comparison between sharded and unsharded model to not match.
I removed this condition so that we skip only for meta devices.

https://www.internalfb.com/intern/test/562950052233328

Differential Revision: D81039244
@meta-cla meta-cla bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Aug 26, 2025
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D81039244

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants