uviol

/

scat-krea2-v1

like 6

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-Turbo,krea/Krea-2-Raw", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("uviol/scat-krea2-v1")

prompt = "shitting, defecating, shit, feces"
image = pipe(prompt).images[0]