from torch._subclasses.fake_tensor import FakeTensorMode with FakeTensorMode(): pass Traceback (most recent call last): File "/data/users/weif/torchtitan/test_fake ...