makriot/

flux-calibri

import torch
from diffusers import DiffusionPipeline
        
device = "cuda" if torch.cuda.is_available() else "cpu"
dtype = torch.bfloat16
model_path = "makriot/flux-calibri"

pipeline = DiffusionPipeline.from_pretrained(
    model_path,
    torch_dtype=dtype
).to(device)

prompts = [
    "a futuristic city at sunset",
    "a cute robot playing guitar"
]

images = pipeline(
    prompts,
    num_inference_steps=15,
    guidance_scale=3.5,
    height=1024,
    width=1024,
).images