How to use argmaxinc/mlx-stable-diffusion-3-medium with DiffusionKit:
# Pipeline for Stable Diffusion 3 from diffusionkit.mlx import DiffusionPipeline pipeline = DiffusionPipeline( shift=3.0, use_t5=False, model_version=argmaxinc/mlx-stable-diffusion-3-medium, low_memory_mode=True, a16=True, w16=True, )
# Image Generation HEIGHT = 512 WIDTH = 512 NUM_STEPS = 50 CFG_WEIGHT = 5 image, _ = pipeline.generate_image( "a photo of a cat", cfg_weight=CFG_WEIGHT, num_steps=NUM_STEPS, latent_size=(HEIGHT // 8, WIDTH // 8), )
How to use argmaxinc/mlx-stable-diffusion-3-medium with MLX:
# Download the model from the Hub pip install huggingface_hub[hf_xet] huggingface-cli download --local-dir mlx-stable-diffusion-3-medium argmaxinc/mlx-stable-diffusion-3-medium
The community tab is the place to discuss and collaborate with the HF community!
