fix: replace deprecated torch.cuda.amp with torch.amp#445

haoyu-haoyu wants to merge 1 commit intoRosettaCommons:mainRosettaCommons/RFdiffusion:mainfrom

haoyu-haoyu:fix/replace-deprecated-torch-cuda-amphaoyu-haoyu/RFdiffusion:fix/replace-deprecated-torch-cuda-ampCopy head branch name to clipboard