cp /usr/local/lib/python3.10/dist-packages/flash_mla/flash_mla_interface.py /usr/local/lib/python3.10/dist-packages/flash_mla/flash_mla_interface.py.org
cp src/flash_mla_interface_torch.py /usr/local/lib/python3.10/dist-packages/flash_mla/flash_mla_interface.py