Release the official PyTorch code of SiT! Check it out here SiT codebase