DiffusionKit How to use argmaxinc/mlx-FLUX.1-schnell with DiffusionKit:
# Pipeline for Flux
from diffusionkit.mlx import FluxPipeline
pipeline = FluxPipeline(
shift=1.0,
model_version=argmaxinc/mlx-FLUX.1-schnell,
low_memory_mode=True,
a16=True,
w16=True,
)
# Image Generation
HEIGHT = 512
WIDTH = 512
NUM_STEPS = 4
CFG_WEIGHT = 0
image, _ = pipeline.generate_image(
"a photo of a cat",
cfg_weight=CFG_WEIGHT,
num_steps=NUM_STEPS,
latent_size=(HEIGHT // 8, WIDTH // 8),
)
