File size: 2,246 Bytes
9d2f9e2 aa92561 40518fd aa92561 a999347 aa92561 13146ea aa92561 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | ---
license: apache-2.0
language:
- en
pipeline_tag: text-to-image
tags:
- comfyui
- diffusion-single-file
base_model:
- lodestones/Chroma1-Flash
base_model_relation: quantized
---
## Update: I have uploaded an updated version of this model, that should further reduce disk size and VRAM usage by ~82 MB. This is because I missed out on compressing a small portion of the model (the `distilled_guidance_layers`) in my original upload. There is <u>no need to download again</u> if you are not having any issues with the older version, but please either update to the latest version of my DF11 custom node fork, or update the patched `pattern_dict` as shown below if you are still using the official node. I apologize for my earlier oversight.
For more information (including how to compress models yourself), check out https://huggingface.co/DFloat11 and https://github.com/LeanModels/DFloat11
Feel free to request for other models for compression as well, although compressing models that do not use the Flux architecture might be slightly tricky for me.
### How to Use
#### ComfyUI
Install my own fork of the DF11 ComfyUI custom node: https://github.com/mingyi456/ComfyUI-DFloat11-Extended. After installing the DF11 custom node, use the provided workflow [json](Chroma1-Flash-DF11-workflow.json), or simply replace the "Load Diffusion Model" node of an existing Chroma workflow with the "DFloat11 Model Loader" node. If you run into any issues, feel free to leave a comment. The workflow is also embedded in the below [png](Chroma1-Flash-DF11-workflow.png) image.
#### `diffusers`
Refer to this [model](https://huggingface.co/mingyi456/Chroma1-Flash-DF11) instead.
### Compression Details
This is the `pattern_dict` for compressing Chroma-based models in ComfyUI:
```python
pattern_dict_comfyui = {
r"distilled_guidance_layer\.layers\.\d+": (
"in_layer",
"out_layer"
),
r"double_blocks\.\d+": (
"img_attn.qkv",
"img_attn.proj",
"img_mlp.0",
"img_mlp.2",
"txt_attn.qkv",
"txt_attn.proj",
"txt_mlp.0",
"txt_mlp.2",
),
r"single_blocks\.\d+": (
"linear1",
"linear2",
),
}
``` |