File size: 1,287 Bytes
c5202e2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | ---
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).
|