Imported from https://huggingface.co/Sebas6k/DiT_weights