import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("makriot/flux-calibri", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
Calibri Flux
To run the model:
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
- Downloads last month
- 3