Views
No views yet
1import torch
2
3from diffusers import ChromaTransformer2DModel
4from diffusers import ChromaPipeline
5
6from transformers import T5EncoderModel, T5Tokenizer
7
8chroma_transformer = ChromaTransformer2DModel.from_single_file("chroma-unlocked-v35.safetensors", torch_dtype=torch.bfloat16)
9
10text_encoder = T5EncoderModel.from_pretrained("black-forest-labs/FLUX.1-schnell", subfolder="text_encoder_2", torch_dtype=torch.bfloat16)
11tokenizer = T5Tokenizer.from_pretrained("black-forest-labs/FLUX.1-schnell", subfolder="tokenizer_2")
12pipe = ChromaPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", transformer=chroma_transformer, text_encoder=text_encoder, tokenizer=tokenizer, torch_dtype=torch.bfloat16)
13
14pipe.save_pretrained("Chroma-v36-dc-diffusers")