Skip to content
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.

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

[PT-D] Relaxed contract to allow Sequence[nn.Module] #127773

Closed
wants to merge 10 commits into from
Prev Previous commit
Update on "[PT-D] Relaxed contract to allow Sequence[nn.Module]"
This PR relaxes `contract` to allow the 1st argument to be `Sequence[nn.Module]` instead of strictly `nn.Module`. This is required for the next PR, which allows `fully_shard` to take in `List[nn.Module]`.

cc mrshenli pritamdamania87 zhaojuanmao satgera gqchen aazzolini osalpekar jiayisuse H-Huang kwen2501 penguinwu fegin XilunWu wanchaol fduwjj wz337 tianyu-l wconstab yf225 chauhang d4l3k

[ghstack-poisoned]
  • Loading branch information
awgu committed Jul 15, 2024
commit d1cd2039892c264684b5d33286b0b914a157a8c3

This merge commit was added into this branch cleanly.

There are no new changes to show, but you can still view the diff.