A copy of https://huggingface.co/spaces/KyanChen/FunSR/tree/main/pretrain