Views
No views yet
1# !pip install diffusers transformers
2from diffusers import DiffusionPipeline
3
4model_id = "CompVis/ldm-text2im-large-256"
5
6# load model and scheduler
7ldm = DiffusionPipeline.from_pretrained(model_id)
8
9# run pipeline in inference (sample random noise and denoise)
10prompt = "A painting of a squirrel eating a burger"
11images = ldm([prompt], num_inference_steps=50, eta=0.3, guidance_scale=6).images
12
13# save images
14for idx, image in enumerate(images):
15 image.save(f"squirrel-{idx}.png")


