transformers
Make vilt, switch_transformers compatible with model parallelism
#22703

Merged

Make vilt, switch_transformers compatible with model parallelism #22703

sgugger merged 2 commits into huggingface:main from Xrenya:parallelism

Update modeling_vilt.py

1d6eba9b

Update modeling_switch_transformers.py

187a7baf

sgugger approved these changes on 2023-04-11

Xrenya closed this 3 years ago

Xrenya reopened this 3 years ago

sgugger merged 95e70575 into main 3 years ago

Reviewers

sgugger

Assignees

No one assigned

Labels

None yet

Milestone

No milestone