Latent Diffusion Model Training with PyTorch FSDP