data preparation for ESM3