Instructions to use Kwai-Kolors/Kolors-diffusers with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use Kwai-Kolors/Kolors-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("Kwai-Kolors/Kolors-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
Remove `torch_dtype` from config
Browse files- text_encoder/config.json +0 -1
text_encoder/config.json
CHANGED
|
@@ -40,7 +40,6 @@
|
|
| 40 |
"rmsnorm": true,
|
| 41 |
"seq_length": 32768,
|
| 42 |
"tie_word_embeddings": false,
|
| 43 |
-
"torch_dtype": "float16",
|
| 44 |
"transformers_version": "4.41.2",
|
| 45 |
"use_cache": true,
|
| 46 |
"vocab_size": 65024
|
|
|
|
| 40 |
"rmsnorm": true,
|
| 41 |
"seq_length": 32768,
|
| 42 |
"tie_word_embeddings": false,
|
|
|
|
| 43 |
"transformers_version": "4.41.2",
|
| 44 |
"use_cache": true,
|
| 45 |
"vocab_size": 65024
|