dit_model/diffusion_pytorch_model-00002-of-00002.safetensors