import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("AcademiaSD/Krea-2-NF4-for-LoRA-Training", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Krea 2 — NF4 for LoRA training
NF4 (4-bit) version of Krea 2 Raw used by AcademiaSD LoRAlab Trainer Studio to train Krea 2 LoRAs on consumer GPUs. The trainer downloads it automatically.
These weights are a derivative of Krea 2 and are distributed under the same Krea 2 Community License. Read it before using the model or the LoRAs trained with it.
- Downloads last month
- 649
Model tree for AcademiaSD/Krea-2-NF4-for-LoRA-Training
Base model
krea/Krea-2-Raw