Instructions to use ethanfel/Krea-2-Base-Diffusers with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use ethanfel/Krea-2-Base-Diffusers with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("ethanfel/Krea-2-Base-Diffusers", 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
| { | |
| "_class_name": "Krea2Transformer2DModel", | |
| "_diffusers_version": "0.39.0.dev0", | |
| "in_channels": 64, | |
| "num_layers": 28, | |
| "attention_head_dim": 128, | |
| "num_attention_heads": 48, | |
| "num_key_value_heads": 12, | |
| "intermediate_size": 16384, | |
| "timestep_embed_dim": 256, | |
| "text_hidden_dim": 2560, | |
| "num_text_layers": 12, | |
| "text_num_attention_heads": 20, | |
| "text_num_key_value_heads": 20, | |
| "text_intermediate_size": 6912, | |
| "num_layerwise_text_blocks": 2, | |
| "num_refiner_text_blocks": 2, | |
| "axes_dims_rope": [ | |
| 32, | |
| 48, | |
| 48 | |
| ], | |
| "rope_theta": 1000.0, | |
| "norm_eps": 1e-05 | |
| } |