use the model consists of roughly 1.4 billion parameters, we strongly recommend running it on GPU.
You can move the generator object to GPU, just like you would in PyTorch.


	Copied
>>> generator.to("cuda")
Now you can use the generator on your text prompt:


	Copied
>>> image = generator().images[0]
The output is by default wrapped into a PIL Image object.
You can save the image by simply calling:


	Copied
>>> image.save("generated_image.png