File size: 4,153 Bytes
c2a6fbc 0b574f5 c2a6fbc 4565d23 c2a6fbc 5e6a69d a07016a 1e8e3e4 2155618 1e8e3e4 a07016a c2a6fbc 0b574f5 c2a6fbc 4565d23 c2a6fbc 9ecb3b7 4565d23 c2a6fbc 4565d23 c2a6fbc 4565d23 c2a6fbc 4565d23 c2a6fbc |
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 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
---
base_model: black-forest-labs/FLUX.1-dev
---
*Note that all these models are derivatives of black-forest-labs/FLUX.1-dev and therefore covered by the
[FLUX.1 [dev] Non-Commercial License](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md) license.*
*Some models are derivatives of finetunes, and are included with the permission of the finetuner*
# Optimised Flux GGUF models
A collection of GGUF models using mixed quantization (different layers quantized to different precision to optimise fidelity v. memory).
They were created using the [convert.py script](https://github.com/chrisgoringe/mixed-gguf-converter).
They can be loaded in ComfyUI using the [ComfyUI GGUF Nodes](https://github.com/city96/ComfyUI-GGUF). Just put the gguf files in your
models/unet directory.
## Naming convention (mx for 'mixed')
[original_model_name]_mxN_N.gguf
where N_N is the average number of bits per parameter.
## Good choices to start with
```
- 9_2 is a good choice for 16 GB cards
- 6_9 just fits on a 12 GB card
- 5_9 is comfortable on 12 GB cards
```
## How is this optimised?
The process for optimisation is as follows:
- 240 prompts used for flux images popular at civit.ai were run through the full Flux.1-dev model with randomised resolution and step count.
- For a randomly selected step in the inference, the hidden states before and after the layer stack were captured.
- For each layer in turn, and for each of the Q8_0, Q5_1 and Q4_1 quantizations:
- A single layer was quantized
- The initial hidden states were processed by the modified layer stack
- The error (MSE) in the final hidden state was calculated
- This gives a 'cost' for each possible layer quantization
- An optimised quantization is one that gives the desired reduction in size for the smallest total cost
- A series of recipies for optimization have been created from the calculated costs
- the various 'in' blocks, the final layer blocks, and all normalization scale parameters are stored in float32
## Also note
- Tests on using bitsandbytes quantizations showed they did not perform as well as the equivalent sized GGUF quants
- Different quantizations of different parts of a layer gave significantly worse results
- Leaving bias in 16 bit made no relevant difference
- Costs were evaluated for the original Flux.1-dev model. They are assumed to be essentially the same for finetunes
## Details
The optimisation recipes are as follows (layers 0-18 are the double_block_layers, 19-56 are the single_block_layers)
```python
CONFIGURATIONS = {
"9_6" : {
'casts': [
{'layers': '0-10', 'castto': 'BF16'},
{'layers': '11-14, 54', 'castto': 'Q8_0'},
{'layers': '15-36, 39-53, 55', 'castto': 'Q5_1'},
{'layers': '37-38, 56', 'castto': 'Q4_1'},
]
},
"9_2" : {
'casts': [
{'layers': '0-8, 10, 12', 'castto': 'BF16'},
{'layers': '9, 11, 13-21, 49-54', 'castto': 'patch:flux1-dev-Q6_K.gguf'},
{'layers': '22-34, 41-48, 55', 'castto': 'patch:flux1-dev-Q5_K_S.gguf'},
{'layers': '35-40', 'castto': 'patch:flux1-dev-Q4_K_S.gguf'},
{'layers': '56', 'castto': 'Q4_1'},
]
},
"8_4" : {
'casts': [
{'layers': '0-4, 10', 'castto': 'BF16'},
{'layers': '5-9, 11-14', 'castto': 'Q8_0'},
{'layers': '15-35, 41-55', 'castto': 'Q5_1'},
{'layers': '36-40, 56', 'castto': 'Q4_1'},
]
},
"7_4" : {
'casts': [
{'layers': '0-2', 'castto': 'BF16'},
{'layers': '5, 7-12', 'castto': 'Q8_0'},
{'layers': '3-4, 6, 13-33, 42-55', 'castto': 'Q5_1'},
{'layers': '34-41, 56', 'castto': 'Q4_1'},
]
},
"5_9" : {
'casts': [
{'layers': '0-25, 27-28, 44-54', 'castto': 'Q5_1'},
{'layers': '26, 29-43, 55-56', 'castto': 'Q4_1'},
]
},
"5_1" : {
'casts': [
{'layers': '0-56', 'castto': 'Q4_1'},
]
},
}
``` |