Skip to content

Images

text to image

Real SD-Turbo output (4 steps) — "a robot arm at a sunlit kitchen counter".

an image tile resolving from noise

Text to image with any of the 100+ image pipelines (Stable Diffusion, SDXL, Flux, Kandinsky, PixArt, Sana, ...).

from strands_diffusers import use_diffusers

use_diffusers(
    action="run",
    pipeline="StableDiffusionPipeline",
    model="stabilityai/stable-diffusion-2-1",
    parameters={"prompt": "a robot arm in a kitchen", "num_inference_steps": 25},
)
# -> artifacts: ['/tmp/strands_diffusers/image_*.png']

Image to image

Pass an input image (path, URL, or base64) - it is coerced to PIL automatically.

use_diffusers(
    action="run",
    pipeline="StableDiffusionImg2ImgPipeline",
    model="stabilityai/stable-diffusion-2-1",
    parameters={"prompt": "make it watercolor", "image": "input.png",
                "strength": 0.6},
)

Inpainting

use_diffusers(
    action="run",
    pipeline="StableDiffusionInpaintPipeline",
    model="runwayml/stable-diffusion-inpainting",
    parameters={"prompt": "a cat", "image": "photo.png", "mask_image": "mask.png"},
)

Batched generation

num_images_per_prompt > 1 saves every image with a collision-free name.

use_diffusers(action="run", pipeline="StableDiffusionPipeline", model="...",
              parameters={"prompt": "robot", "num_images_per_prompt": 4})
# -> 4 distinct .png artifacts

Find an image pipeline

use_diffusers(action="modalities")["data"]["image"]   # 100+ classes