Skip to content

Commit 7635f7e

Browse files
authored
enhance flux doc (#967)
1 parent c4ef9a8 commit 7635f7e

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

auto_round/compressors/diffusion/README.md

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,15 @@ This feature is experimental and may be subject to changes, including potential
44

55
## Quantization
66

7+
Quantization for diffusion models is limited:
8+
9+
1. Only transformer module of diffusion models will be quantized..
10+
2. Loading quantized model is not supported yet, so please use `fake` format for quantization.
11+
3. Calibration dataset only supports `coco2014` and user customized `.tsv` file.
12+
13+
714
### API Usage (CPU/GPU) Recommended
815

9-
By default, AutoRoundDiffusion only quantizes the transformer module of diffusion models and uses `COCO2014 captions` for calibration.
1016

1117
```python
1218
import torch

0 commit comments

Comments
 (0)