Hi, using a MacBook Pro 16" 2019, and I received that AssertionErrror when running example.py I have already used torch.load with map_location=torch.device('cpu') but still get that error message. What should I do?