Back to Transformers

Quantization

docs/source/en/main_classes/quantization.md

5.8.02.4 KB
Original Source
<!--Copyright 2023 The HuggingFace Team. All rights reserved. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. ⚠️ Note that this file is in Markdown but contain specific syntax for our doc-builder (similar to MDX) that may not be rendered properly in your Markdown viewer. -->

Quantization

Quantization techniques reduce memory and computational costs by representing weights and activations with lower-precision data types like 8-bit integers (int8). This enables loading larger models you normally wouldn't be able to fit into memory, and speeding up inference. Transformers supports the AWQ and GPTQ quantization algorithms and it supports 8-bit and 4-bit quantization with bitsandbytes.

Quantization techniques that aren't supported in Transformers can be added with the [HfQuantizer] class.

<Tip>

Learn how to quantize models in the Quantization guide.

</Tip>

QuantoConfig

[[autodoc]] QuantoConfig

AqlmConfig

[[autodoc]] AqlmConfig

VptqConfig

[[autodoc]] VptqConfig

AwqConfig

[[autodoc]] AwqConfig

EetqConfig

[[autodoc]] EetqConfig

GPTQConfig

[[autodoc]] GPTQConfig

BitsAndBytesConfig

[[autodoc]] BitsAndBytesConfig

HfQuantizer

[[autodoc]] quantizers.base.HfQuantizer

HiggsConfig

[[autodoc]] HiggsConfig

HqqConfig

[[autodoc]] HqqConfig

MetalConfig

[[autodoc]] MetalConfig

Mxfp4Config

[[autodoc]] Mxfp4Config

FbgemmFp8Config

[[autodoc]] FbgemmFp8Config

CompressedTensorsConfig

[[autodoc]] CompressedTensorsConfig

TorchAoConfig

[[autodoc]] TorchAoConfig

BitNetQuantConfig

[[autodoc]] BitNetQuantConfig

SpQRConfig

[[autodoc]] SpQRConfig

FineGrainedFP8Config

[[autodoc]] FineGrainedFP8Config

QuarkConfig

[[autodoc]] QuarkConfig

FourOverSixConfig

[[autodoc]] FourOverSixConfig

FPQuantConfig

[[autodoc]] FPQuantConfig

AutoRoundConfig

[[autodoc]] AutoRoundConfig

SinqConfig

[[autodoc]] SinqConfig