v8.3.1: Fix torch deprecation warning

github-actions released this 30 Sep 19:08

· 73 commits to main since this release

torch.cuda.amp is deprecated (Pytorch 2.4). This PR updates shims pytorch.py to use torch.amp.autocast instead of torch.cuda.amp.autocast.

Thanks to @Atlogit for the patch.

Contributors

Atlogit

Assets 19