transformers

Fix the key name for _load_rng_state under torch.cuda

#36138

Merged