Method __init__

src/diffusers_data_pipeline.py:261–263

(self, prompt, num_samples)

    "A simple dataset to prepare the prompts to generate class images on multiple GPUs."

    def __init__(self, prompt, num_samples):
        self.prompt = prompt
        self.num_samples = num_samples

    def __len__(self):
        return self.num_samples