Features

• Command line interface
• UI Features. Query models with multiple parametrs
  • Text prompting (text2img)
  • Image based prompting (img2img)
  • Editor (for inpainting and outpainting)
  • Latent space exploration
• Experimentation tools
  • Save intermediate images in the sampling loop
  • Prompt recommendation tools
  • Model explanations
  • Curation/sharing experiment results

import os
from dotenv import load_dotenv
from peacasso.generator import ImageGenerator
from peacasso.datamodel import GeneratorConfig

token = os.environ.get("HF_API_TOKEN")
gen = ImageGenerator(token=token)
prompt = "A sea lion wandering the streets of post apocalyptic London"

prompt_config = GeneratorConfig(
    prompt=prompt,
    num_images=3,
    width=512,
    height=512,
    guidance_scale=7.5,
    num_inference_steps=50,
    mode="prompt",  # prompt, image
    return_intermediates=True, # return intermediate images in the generate dict response
)

result = gen.generate(prompt_config)
for i, image in enumerate(result["images"]):
    image.save(f"image_{i}.png")