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

Add Auto Device Map option for BERT Models #26176

Closed
wants to merge 1 commit into from

add: only Self attention and embeddings to _no_split_modules

e3f8e9b
Select commit
Loading
Failed to load commit list.
Closed

Add Auto Device Map option for BERT Models #26176

add: only Self attention and embeddings to _no_split_modules
e3f8e9b
Select commit
Loading
Failed to load commit list.
CircleCI Checks / setup_and_quality failed Sep 15, 2023 in 5m 9s

Workflow: setup_and_quality