--- license: other license_name: krea-2-community license_link: https://www.krea.ai/krea-2-licensing pipeline_tag: text-to-image library_name: diffusers base_model: krea/Krea-2-Raw base_model_relation: finetune tags: - text-to-image - flow-matching - krea --- # Krea 2 Base — Diffusers (working conversion) A **diffusers-loadable** conversion of the Krea 2 **Base/RAW** checkpoint that loads cleanly with `diffusers.Krea2Transformer2DModel` / `Krea2Pipeline` (transformer keys renamed to the diffusers layout, per-block modulation table reshaped, and the two final-layer `up`/`down` weights dropped to match the diffusers Krea2 port). The `vae`, `text_encoder`, `tokenizer` and `scheduler` are the standard diffusers components. Intended for **fine-tuning / LoRA training** (train on Base/RAW, run inference on Krea 2 Turbo). ```python import torch from diffusers import Krea2Pipeline pipe = Krea2Pipeline.from_pretrained("ethanfel/Krea-2-Base-Diffusers", torch_dtype=torch.bfloat16) ``` In ai-toolkit (krea2 branch), set `model.name_or_path: "ethanfel/Krea-2-Base-Diffusers"` and `model.arch: "krea_2"`. Converted with `scripts/convert_krea2_community_to_diffusers.py`. Weights are redistributed under the [Krea 2 Community License](https://www.krea.ai/krea-2-licensing).