Text-to-Image
Diffusers
Safetensors
English
Krea2Pipeline
image-generation
krea2
orbitquant
w4a4
4-bit precision
quantized
8-bit precision
Instructions to use WaveCut/Krea-2-Turbo-OrbitQuant-W4A4 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use WaveCut/Krea-2-Turbo-OrbitQuant-W4A4 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("WaveCut/Krea-2-Turbo-OrbitQuant-W4A4", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
File size: 1,102 Bytes
96430e9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | {
"model": "WaveCut/Krea-2-Turbo-OrbitQuant-W4A4",
"revision": "49e89c37f51cb394b4ee199196d9df8bcd042475",
"gpu": "NVIDIA A40",
"capability": [
8,
6
],
"torch": "2.8.0+cu128",
"layer": "transformer_blocks.0.ff.gate",
"shape": {
"rows": 16418,
"in_features": 6144,
"out_features": 16384
},
"int_mm_chunked": {
"samples_seconds": [
0.021558222826570272,
0.020911033730953932
],
"median_seconds": 0.021234628278762102,
"peak_allocated_mib": 7793.1279296875,
"activation_backend": "triton_cuda_packed_w4",
"native_w4a4_enabled": true,
"output_mean": 0.001386194839142263,
"output_std": 1.7071678638458252
},
"native_direct_packed": {
"samples_seconds": [
0.057071581948548555,
0.05693301185965538
],
"median_seconds": 0.05700229690410197,
"peak_allocated_mib": 7459.9912109375,
"activation_backend": "triton_cuda_packed_w4",
"native_w4a4_enabled": true,
"output_mean": 0.0013861971674486995,
"output_std": 1.7071678638458252
},
"direct_over_int_mm": 2.684402861015139
}
|