class_docstr: A simple dataset to prepare the prompts to generate class images on multiple GPUs.
code: def __init__(self, prompt, num_samples):
      self.prompt = prompt
      self.num_samples = num_samples
  
imports: from pathlib import Path
  import torch
  from PIL import Image
  from PIL.ImageOps import exif_transpose
  from torch.utils.data import Dataset
  from torchvision import transforms
label_desc: Use image node_id 1 to create a new PromptDataset object from inherited base classes: Dataset with example: obj = PromptDataset(prompt, num_samples)