pith. sign in

arxiv: 2412.18653 · v1 · pith:XHRWC26Inew · submitted 2024-12-24 · 💻 cs.CV · cs.AI· cs.LG

1.58-bit FLUX

classification 💻 cs.CV cs.AIcs.LG
keywords fluxmodelgenerationinferencemaintainingreductionwhileaccess
0
0 comments X
read the original abstract

We present 1.58-bit FLUX, the first successful approach to quantizing the state-of-the-art text-to-image generation model, FLUX.1-dev, using 1.58-bit weights (i.e., values in {-1, 0, +1}) while maintaining comparable performance for generating 1024 x 1024 images. Notably, our quantization method operates without access to image data, relying solely on self-supervision from the FLUX.1-dev model. Additionally, we develop a custom kernel optimized for 1.58-bit operations, achieving a 7.7x reduction in model storage, a 5.1x reduction in inference memory, and improved inference latency. Extensive evaluations on the GenEval and T2I Compbench benchmarks demonstrate the effectiveness of 1.58-bit FLUX in maintaining generation quality while significantly enhancing computational efficiency.

This paper has not been read by Pith yet.

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Forward citations

Cited by 5 Pith papers

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. When Do Diffusion Models learn to Generate Multiple Objects?

    cs.CV 2026-04 unverdicted novelty 8.0

    Diffusion models' multi-object generation is limited primarily by scene complexity and held-out combinations rather than imbalance, with counting difficult in low data and compositional generalization collapsing as mo...

  2. A Frame is Worth One Token: Efficient Generative World Modeling with Delta Tokens

    cs.CV 2026-04 conditional novelty 7.0

    Delta tokens compress VFM feature differences into single tokens, enabling a lightweight generative world model that predicts diverse futures with far lower compute than existing approaches.

  3. RT-Lynx: Putting the GEMM Sparsity In a Right Way for Diffusion Models

    cs.LG 2026-05 unverdicted novelty 6.0

    RT-Lynx shifts DiT sparsity from weights to activations, reports up to 1.55x linear-layer speedup while preserving generation quality across multiple diffusion models.

  4. When Do Diffusion Models learn to Generate Multiple Objects?

    cs.CV 2026-04 unverdicted novelty 6.0

    Using the mosaic controlled dataset framework, experiments show scene complexity dominates over concept imbalance in diffusion model failures for multi-object generation, with counting especially hard in low-data regi...

  5. Frequency-Aware Flow Matching for High-Quality Image Generation

    cs.CV 2026-04 unverdicted novelty 6.0

    FreqFlow introduces frequency-aware conditioning and a two-branch architecture to flow matching, reaching FID 1.38 on ImageNet-256 and outperforming DiT and SiT.