Fix gradient checkpointing bug in switch transformer#22081

gante merged 1 commit into

huggingface:mainhuggingface/transformers:mainfrom

KMFODA:fix_gradient_checkpointing_switchtransformersKMFODA/transformers:fix_gradient_checkpointing_switchtransformersCopy head branch name to clipboard