class_docstr: A simple dataset to prepare the prompts to generate class images on multiple GPUs.
code: def __getitem__(self, index):
      example = {}
      example["prompt"] = self.prompt
      example["index"] = index
      return example
  
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 3 for calling the PromptDataset obj's underlying member method code with example usage: obj.__getitem__(index) and returns: example