transformers

12962fe8 - Fix the key name for _load_rng_state under torch.cuda (#36138)

Fix the key name for _load_rng_state under torch.cuda (#36138) fix load key name for _load_rng_state under torch.cuda Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>