ny PyTorch module:


	Copied
pipeline = pipeline.to("cuda")
To make sure you can use the same image and improve on it, use a Generator and set a seed for reproducibility:


	Copied
generator = torch.Generator("cuda").manual_seed(0)
Now you can generate an image:


	Copied
image = pipeline(prompt, generator=generator).images[0]
image

This process took ~30 seconds on a T4 GPU (it might be faster if your allocated GPU is better than a T4). By def