imnotednamode commited on
Commit
01d74cb
·
verified ·
1 Parent(s): bd3b261

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -3
README.md CHANGED
@@ -1,3 +1,24 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model:
4
+ - lodestones/Chroma
5
+ ---
6
+
7
+ Generated with the following script
8
+
9
+ ```py
10
+ import torch
11
+
12
+ from diffusers import ChromaTransformer2DModel
13
+ from diffusers import ChromaPipeline
14
+
15
+ from transformers import T5EncoderModel, T5Tokenizer
16
+
17
+ chroma_transformer = ChromaTransformer2DModel.from_single_file("chroma-unlocked-v35.safetensors", torch_dtype=torch.bfloat16)
18
+
19
+ text_encoder = T5EncoderModel.from_pretrained("black-forest-labs/FLUX.1-schnell", subfolder="text_encoder_2", torch_dtype=torch.bfloat16)
20
+ tokenizer = T5Tokenizer.from_pretrained("black-forest-labs/FLUX.1-schnell", subfolder="tokenizer_2", torch_dtype=torch.bfloat16)
21
+ pipe = ChromaPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", transformer=chroma_transformer, text_encoder=text_encoder, tokenizer=tokenizer, torch_dtype=torch.bfloat16)
22
+
23
+ pipe.save_pretrained("Chroma-v36-dc-diffusers")
24
+ ```