mages[0].save("variation_image.png")
Optionally, you can also pass a prompt to pipe such as:


	Copied
prompt = "A fantasy landscape, trending on artstation"

images = pipe(init_image, prompt=prompt).images
images[0].save("variation_image_two.png")

Memory optimization
	
If you are short on GPU memory, you can enable smart CPU offloading so that models that are not needed
immediately for a computation can be offloaded to CPU:


	Copied
from dif