conda env create -f environment.yml
conda activate DiT