How to use from the
Use from the
Diffusers library
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline

# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("krea/Krea-2-Raw", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("aimalias/p3n1s-krea")

prompt = "TOK"
image = pipe(prompt).images[0]

Krea 2 DreamBooth LoRA - aimalias/p3n1s-krea

Model description

These are aimalias/p3n1s-krea DreamBooth LoRA weights, trained on krea/Krea-2-Raw.

The weights were trained using DreamBooth with the Krea 2 diffusers trainer.

Krea 2 ships as two checkpoints: RAW (the non-distilled base you fine-tune on) and Turbo (an 8-step distilled checkpoint for fast, high-quality inference). Train your LoRA on RAW and run it on Turbo — LoRAs trained on RAW express strongly on Turbo.

Trigger words

You should use TOK to trigger the image generation.

Download model

Download the *.safetensors LoRA in the Files & versions tab.

Use it with the 🧨 diffusers library

>>> import torch
>>> from diffusers import Krea2Pipeline

>>> # Load the LoRA onto Krea 2 Turbo (the distilled inference model)
>>> pipe = Krea2Pipeline.from_pretrained("krea/Krea-2-Turbo", torch_dtype=torch.bfloat16).to("cuda")
>>> pipe.load_lora_weights("aimalias/p3n1s-krea")

>>> # Turbo recipe: 8 steps, no classifier-free guidance
>>> image = pipe("TOK", num_inference_steps=8, guidance_scale=0.0).images[0]
>>> image.save("output.png")

For more details, including weighting, merging and fusing LoRAs, check the documentation on loading LoRAs in diffusers

Intended uses & limitations

How to use

# TODO: add an example code snippet for running this diffusion pipeline

Limitations and bias

[TODO: provide examples of latent issues and potential remediations]

Training details

[TODO: describe the data used to train the model]

Downloads last month
-
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for aimalias/p3n1s-krea

Base model

krea/Krea-2-Raw
Adapter
(10)
this model