pts for training PromptDataset: generates the prompt embeddings to generate the class images If you enabled prior preservation loss, the class images are generated here:   Copied sample_dataset = PromptDataset(args.class_prompt, num_new_images)
sample_dataloader = torch.utils.data.DataLoader(sample_dataset, batch_size=args.sample_batch_size)

sample_dataloader = accelerator.prepare(sample_dataloader)
pipeline.to(accelerator.device)

for example