when we are run inference_codeformer.py on macbook M1 get error Segmentation fault: 11

[i51888]

$ python inference_codeformer.py --input_path inputs/cropped_faces/0143.png
Segmentation fault: 11

we are found this function has problem:
# ------------------ set up CodeFormer restorer -------------------
net = ARCH_REGISTRY.get('CodeFormer')(dim_embd=512, codebook_size=1024, n_head=8, n_layers=9,
connect_list=['32', '64', '128', '256']).to(device)
i get the device is mps, but the net hasn't any return.