|
|
--- |
|
|
license: apache-2.0 |
|
|
base_model: |
|
|
- lodestones/Chroma |
|
|
library_name: diffusers |
|
|
--- |
|
|
|
|
|
Generated with the following script |
|
|
|
|
|
```py |
|
|
import torch |
|
|
|
|
|
from diffusers import ChromaTransformer2DModel |
|
|
from diffusers import ChromaPipeline |
|
|
|
|
|
from transformers import T5EncoderModel, T5Tokenizer |
|
|
|
|
|
chroma_transformer = ChromaTransformer2DModel.from_single_file("chroma-unlocked-v35.safetensors", torch_dtype=torch.bfloat16) |
|
|
|
|
|
text_encoder = T5EncoderModel.from_pretrained("black-forest-labs/FLUX.1-schnell", subfolder="text_encoder_2", torch_dtype=torch.bfloat16) |
|
|
tokenizer = T5Tokenizer.from_pretrained("black-forest-labs/FLUX.1-schnell", subfolder="tokenizer_2") |
|
|
pipe = ChromaPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", transformer=chroma_transformer, text_encoder=text_encoder, tokenizer=tokenizer, torch_dtype=torch.bfloat16) |
|
|
|
|
|
pipe.save_pretrained("Chroma-v36-dc-diffusers") |
|
|
``` |
