mingyi456 commited on
Commit
f4db570
·
verified ·
1 Parent(s): a6a0faf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -8,7 +8,7 @@ language:
8
  pipeline_tag: text-to-image
9
  library_name: diffusers
10
  ---
11
- For more information (including how to compress models yourself), check out https://huggingface.co/DFloat11
12
 
13
  This is my first time using DF11 to compress a model outside the Flux architecture. The process for compressing Flux-based models is much more straightforward as compared to other architectures because the compression code requires a `pattern_dict` as input, but the original [example code](https://github.com/LeanModels/DFloat11/tree/master/examples/compress_flux1) only provides it for Flux, which meant I had to learn the notation myself and modify it to fit other models. At least Chroma is just a pruned version of Flux, so it was relatively simple to derive the correct `pattern_dict` this time. Do let me know if you run into any problems.
14
 
@@ -65,7 +65,7 @@ pattern_dict = {
65
  torch_dtype=torch.bfloat16
66
  ).to(torch.bfloat16)
67
 
68
- pipe = Cosmos2TextToImagePipeline.from_pretrained(
69
  "lodestones/Chroma1-HD",
70
  transformer=transformer,
71
  torch_dtype=torch.bfloat16
 
8
  pipeline_tag: text-to-image
9
  library_name: diffusers
10
  ---
11
+ For more information (including how to compress models yourself), check out https://huggingface.co/DFloat11 and https://github.com/LeanModels/DFloat11
12
 
13
  This is my first time using DF11 to compress a model outside the Flux architecture. The process for compressing Flux-based models is much more straightforward as compared to other architectures because the compression code requires a `pattern_dict` as input, but the original [example code](https://github.com/LeanModels/DFloat11/tree/master/examples/compress_flux1) only provides it for Flux, which meant I had to learn the notation myself and modify it to fit other models. At least Chroma is just a pruned version of Flux, so it was relatively simple to derive the correct `pattern_dict` this time. Do let me know if you run into any problems.
14
 
 
65
  torch_dtype=torch.bfloat16
66
  ).to(torch.bfloat16)
67
 
68
+ pipe = ChromaPipeline.from_pretrained(
69
  "lodestones/Chroma1-HD",
70
  transformer=transformer,
71
  torch_dtype=torch.bfloat16