metadata
license: apache-2.0
base_model:
- lodestones/Chroma
library_name: diffusers
Generated with the following script
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")