Instructions to use krea/Krea-2-Turbo with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use krea/Krea-2-Turbo with Diffusers:
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", dtype=torch.bfloat16, device_map="cuda") prompt = "A small, dark-colored cat is captured mid-stride, walking down the center of a narrow, abandoned street. The street is paved and appears cracked and worn. On either side of the street are tall, dilapidated buildings with visible brickwork and windows. A street lamp stands on the right side. The entire image is rendered in a monochromatic blue, with a distinct halftone dot pattern overlaying the scene, giving it a retro or printed appearance. The focus is soft, and the lighting is diffused, creating a hazy, atmospheric effect. The perspective is from ground level, looking down the length of the street, which narrows into the distance., halftone texture" image = pipe(prompt).images[0] - Inference
- Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
VRAM
How much GPU Vram needed to run Trubo model for a HD image?
any minimum hardware spec?
based on the size of the weights of the diffusion model it should be somewhere around 16Gb of minimum VRAM =( less than it the only way is use a GGUF or FP8 version
just use wan2gp, i did a 1080p image in less than 2minutes using just 5vram
if you run it in things like comfyui, they handle the ram swapping intelligently. a 10gb-ish, even bigger, fp8 or gguf vrsion of most image models does ok, if relatively slow on my old 5gb vram laptop (4050).
MAybe not flux.2 dev (30gb) or hunyuan 3 (80gb).
I am running the RTX A2000 8GB GPU using the default model configs 1024x1024
I was getting OOM errors until I did the following.
I got 1024x1024 Image generation working on the 8 GB VRAM class RTX A2000 setup in NORMAL_VRAM with DynamicVRAM enabled.
Docker rebuild details:
CUDA base image change to 12.6.3 cudnn runtime
PyTorch wheel index change to cu126
Installed torch, torchvision, torchaudio versions
CUDA-specific runtime libraries pulled by torch:
cuDNN, cuBLAS, CUDA runtime/NVRTC, cuFFT, cuRAND, cuSOLVER, cuSPARSE, cuSPARSELT, NCCL, NVJITLINK, NVTX, cuFile