params = { "latent_dim": 1024, "batch_size": 256, "num_diffusion_steps": 1000, "diff