CoolFace
Modelpublic

SceneWorks/krea-2-raw-mlx

sourceHugging Faceotherupdated 3mo agoView on Hugging Face
1likes
Model Card

Krea 2 Raw — MLX (turnkey: bf16 / Q8 / Q4)

On-device, Apple-MLX-ready repack of [krea/Krea-2-Raw](https://huggingface.co/krea/Krea-2-Raw), the undistilled 12B single-stream text-to-image checkpoint from Krea.ai, Inc. — the full classifier-free-guidance base model (and the LoRA-training base) behind the distilled Krea 2 Turbo. This repository is a Derivative prepared for `mlx-gen` (and the SceneWorks worker that embeds it): the weights are group-wise-affine quantized and repacked from the original bf16 diffusers checkpoint so the model loads and runs natively on Apple Silicon with no Python/PyTorch sidecar. The bf16/ tier is the dense original re-layout (max fidelity + the LoRA-training base).

This is not the original checkpoint. For the reference model, training details, and the canonical diffusers / SGLang inference paths, see the upstream card: <https://huggingface.co/krea/Krea-2-Raw>.

Attribution

  • Original model: Krea 2 Raw — © Krea.ai, Inc., released 2026-06-22.
  • Base model: `krea/Krea-2-Raw` (the undistilled base; Krea 2 Turbo is distilled from it).
  • This Derivative: quantized + MLX-repacked by the SceneWorks / mlx-gen project. No retraining or fine-tuning was performed — only numerical quantization and on-disk re-layout.

License

Use of these weights is governed by the Krea 2 Community License Agreement and the Krea Acceptable Use Policy, exactly as for the original model. A copy of the license is included in this repository as `LICENSE.pdf` (also at <https://huggingface.co/krea/Krea-2-Raw/blob/main/LICENSE.pdf>). In the event of any conflict, the Krea Acceptable Use Policy and Krea 2 Community License control.

Deployer obligation (content filtering). The Krea 2 Community License requires anyone who deploys the model to implement content-filtering measures or equivalent review processes appropriate to their use case, to prevent the generation or distribution of unlawful or policy-violating content. If you serve this model, you are responsible for those safeguards. Report harmful, illegal, or policy-violating outputs to safety@krea.ai (potential CSAM is escalated to NCMEC as required by law).

Krea does not claim copyright over generated outputs; users are solely responsible for their inputs and any use of the outputs.

What changed vs. the upstream checkpoint

The conversion is lossy only through quantization — the architecture, tokenizer, scheduler config, and VAE are byte-for-byte the originals.

  • Transformer (DiT) and Qwen3-VL-4B text encoder: for the Q8 / Q4 tiers the linear projection weights are quantized to group-wise affine Q8 / Q4 (group size 64) and repacked into a single .safetensors per stack. Norms, embeddings, modulation tables, and the text-encoder vision tower stay dense. The bf16/ tier keeps every weight dense.
  • VAE (AutoencoderKLQwenImage): copied unchanged (f32).
  • `tokenizer/`, `scheduler/`, `model_index.json`: copied unchanged.

Repository layout

Each tier is a complete, self-contained snapshot you can load directly:

PathQuantizationOn-disk sizeNotes
bf16/none (dense bf16)~35.7 GBMax fidelity; the LoRA-training base.
q8/Q8 (group size 64)~20.6 GBDefault. Near-lossless; needs a 48 GB-class Mac.
q4/Q4 (group size 64)~12.5 GBLighter footprint; mild quality trade-off.
krea-2-raw-mlx/
├── LICENSE.pdf
├── README.md
├── bf16/ { transformer/  text_encoder/  vae/  tokenizer/  scheduler/  model_index.json }
├── q8/   { transformer/  text_encoder/  vae/  tokenizer/  scheduler/  model_index.json }
└── q4/   { transformer/  text_encoder/  vae/  tokenizer/  scheduler/  model_index.json }

Usage

Built for Apple-Silicon inference through mlx-gen's krea_2_raw engine. Point a loader at a tier subdirectory (bf16/, q8/, or q4/); it auto-detects the packed weights. Unlike the CFG-free Turbo, Krea 2 Raw is a true classifier-free-guidance model — run ~52 steps with a real guidance scale (~3.5) and an optional negative prompt. The same bf16/ tier is also the base for Krea 2 LoRA training.

Model details

See the upstream card for the full model overview, capabilities, intended/out-of-scope uses, training-data summary, safety measures, and risk/limitation disclosures: <https://huggingface.co/krea/Krea-2-Raw>.