You can deploy this model and start earning money today!

ethanfel/
Krea-2-Base-Diffusers

Name: Krea-2-Base-Diffusers
Availability: InStock

Views

No views yet

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.