params = { "latent_dim": 1024, "batch_size": 1024, "num_diffusion_steps": 1000, "dif