Instructions to use SceneWorks/chroma1-hd-mlx with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use SceneWorks/chroma1-hd-mlx with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("SceneWorks/chroma1-hd-mlx", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
license: apache-2.0
pipeline_tag: text-to-image
base_model: lodestones/Chroma1-HD
tags:
- mlx
- apple-silicon
- chroma
- flux
- text-to-image
Chroma1-HD — MLX packed tiers (SceneWorks)
Pre-quantized, packed-load MLX tiers of lodestones/Chroma1-HD
for on-device inference in SceneWorks via mlx-gen-chroma.
Chroma1-HD is the high-detail full-CFG FLUX.1-schnell-derived text-to-image DiT (Apache-2.0). This repo
re-hosts it as three self-contained tiers so SceneWorks can load a tier directly with no dense
transient and no in-app quantization (the loader packed-detects group-wise affine weights via
{base}.scales):
| Tier subdir | Transformer weights | Notes |
|---|---|---|
bf16/ |
dense bf16 | verbatim mirror of the source diffusers snapshot |
q8/ |
packed Q8 (group 64) | transformer block Linears only |
q4/ |
packed Q4 (group 64) | transformer block Linears only (default tier) |
Quant scope. Only the DiT transformer/ matmul-heavy block Linears are quantized (the double
blocks' attention + FFN and the single blocks' attention + proj_mlp/proj_out). The transformer's
x_embedder/context_embedder/proj_out and the distilled-guidance Approximator, the shared T5-XXL
text encoder, and the FLUX.1 VAE stay dense in every tier. The Q4/Q8 packing is byte-identical to the
load-time quantization seam (weights cast to bf16 first, MLX group-wise affine at group size 64).
Each tier subdir is a complete diffusers-layout turnkey (transformer/ text_encoder/ vae/ tokenizer/ scheduler/ model_index.json).
License
Apache-2.0, inherited from the upstream model. See LICENSE. Upstream:
lodestones/Chroma1-HD.