 torch_dtype=torch.float16)
>>> pipe = pipe.to("cuda")
Now, let’s define 4 different generators, since we would like to reproduce a certain image. We’ll use seeds 0 to 3 to create our generators.


	Copied
>>> import torch

>>> generator = [torch.Generator(device="cuda").manual_seed(i) for i in range(4)]
Let’s generate 4 images:


	Copied
>>> images = pipe(prompt, generator=generator, num_images_per_prompt=4).images
>>> images

Ok, the last ima