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>
Showing
+1 -1
Please register or sign in to comment