Skip to content

krea-ai/krea-2

Repository files navigation

Krea 2 (K2)

Krea 2 - an image generation model from Krea AI.


API DocsHugging Face (RAW)Hugging Face (TURBO)Technical Blog

k2 banner

k2 banner

This is the official repository for the open version of Krea 2, an image model trained from scratch focused on creative and stylistic exploration. The repository contains inference code and instructions to run the model.

Krea 2 is the most aesthetic open-source image model available. On quality, Krea 2 is the #1 text-to-image model from an independent lab on Artificial Analysis [1].

Krea 2 ships as two models. Krea 2 RAW is the base model. It's a pretrained checkpoint with no distillation, so it's diverse and highly malleable, and it's what you should use for fine-tuning, post-training, and LoRA training. Krea 2 Turbo is an 8-step distilled checkpoint built for fast, high-quality text-to-image.

The two models are designed to work together. You train LoRAs on RAW and apply them on Turbo, and the LoRAs trained on RAW will work well on Turbo. We highly recommend using RAW for training LoRAs and applying them on Turbo for inference.

Setup

uv sync

Both Raw and Turbo safetensor files are available on Hugging Face. After downloading the checkpoints, set the OSS_RAW and OSS_TURBO environment variables to the paths of the downloaded files.

export OSS_RAW=...
export OSS_TURBO=...

Usage

The following commands run inference using the two available checkpoints with recommended settings.

Raw (oss_raw)

The base undistilled model. Use the full sampler with classifier-free guidance: The model has been trained to generate upto 1k resolution.

uv run inference.py "a fox walking in the snow" \
    --checkpoint oss_raw --steps 52 --cfg 3.5

Turbo (oss_turbo)

Distilled for few-step sampling — run with 8 steps and CFG disabled. The model can generate images from 1k ~ 2k resolution.

uv run inference.py "a fox walking in the snow" \
    --checkpoint oss_turbo --steps 8 --cfg 0.0 --mu 1.15 --width 2048 --height 2048

Options

Flag Default Description
prompt (positional) Text prompt to generate from.
--steps 28 Number of denoising steps.
--cfg 4.5 Classifier-free guidance scale (0 disables CFG).
--y1 0.5 Timestep-shift mu at min resolution.
--y2 1.15 Timestep-shift mu at max resolution.
--mu None Pin a constant timestep-shift mu, overriding the resolution-derived value. Recommended 1.15 for oss_turbo.
--width / --height 1024 ~ 2048 Output resolution; padded up to a multiple of 16 if needed.
--num-images 1 Number of images to generate from the prompt.
--seed 0 Base seed; image i uses seed + i.
--checkpoint oss_raw Checkpoint to load (oss_raw, oss_turbo). Defaults to $K2_CHECKPOINT.
--output sample Output filename prefix.

Documentation

Inference

You can run our open source models on the following platforms.

Finetuning Krea 2

For finetuning Krea 2, we highly recommend that you train a LoRA on the Raw model and apply it to the Turbo model for inference. We recommend using the following providers and open source tools for finetuning Krea 2.

FAQ

Which model I should use?

Use the Turbo model for fast inference with high quality results. The Raw model is an undistilled checkpoint without any step / cfg guidance distillation and posttraining. It is a highly finetunable base model that can be used to train LoRAs for the Turbo model as well as posttraining research. In short, TRAIN on Raw and RUN on Turbo.

What license is this model released under?

Both model weights are under our community license with permissive use. To purchase a commercial license, please contact us at opensource@krea.ai.

Citation

@misc{krea-2-2026,
    author={Sangwu Lee, Erwann Millon, Le Zhuo, Matthew Newton, Andrei Filatov, Abhinay Devarinti, Dazhi Zhong, Avram Djordjevic, Gabriel Menezes, Will Beddow, Titus Ebbecke, Mihai Petrescu, Owen Fahey, Gian Saß, Felix Gil, Victor Perez},
    title={{Krea 2}},
    year={2026},
    howpublished={\url{https://www.krea.ai/blog/krea-2-technical-report}},
}

About

Official inference code for Krea 2

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages