Hi, how it is possible to change the code for M1 Gpu's ? The new PyTorch version supports the M1 chip.