 1.4 billion parameters.
You can move the generator object to a GPU, just like you would in PyTorch:   Copied >>> pipeline.to("cuda") Now you can pass a text prompt to the pipeline to generate an image, and then access the denoised image. By default, the image output is wrapped in a PIL.Image object.   Copied >>> image = pipeline("An image of a squirrel in Picasso style").images[0]
>>> image  Save the image by calling save:   Copied >>> image.s