After the models are loaded It get this Error: `module 'torch._C' has no attribute '_cuda_resetPeakMemoryStats'` On a M1 MacBook Pro 32 GB