Fix the key name for _load_rng_state under torch.cuda #36138
muellerzr
merged
2 commits into
huggingface:main
from
wizyoung:fix_load_rng_state
fix load key name for _load_rng_state under torch.cuda
2b69b60f
wizyoung
changed the title 
SunMarc
approved these changes
on 2025-02-13
Merge branch 'main' into fix_load_rng_state
3738e222
muellerzr
approved these changes
on 2025-02-13
Login to write a write a comment.
Login via GitHub