6 lines
202 B
Python
6 lines
202 B
Python
|
from diffusers import DiffusionPipeline
|
||
|
|
|
||
|
|
pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
|
||
|
|
pipeline.to("cuda")
|
||
|
|
pipeline("An image of a squirrel in Picasso style").images[0]
|