yandex-research / vqdm

Official repository for VQDM:Accurate Compression of Text-to-Image Diffusion Models via Vector Quantization paper
Apache License 2.0
15 stars 1 forks source link

Does this work on flux model? #3

Open ninjasaid2k opened 1 week ago

Godofnothing commented 1 week ago

Hi, @ninjasaid2k.

The proposed method is architecture agnostic, so, in principle, it works with any modern diffusion model. Given that the FLUX model is much larger, the benefits from its compression are more pronounced. However, the current implementation is focused towards SDXL-like architectures and adding support for FLUX would require significant effort. We hope to produce VQDM quantized FLUX models in the future.