Files
huggingface--transformers/docs/source/en/serve-cli/serving_optims.md
T
wehub-resource-sync e06fe8e8c6
Secret Leaks / trufflehog (push) Failing after 1s
Build documentation / build (push) Failing after 1s
Build documentation / build_other_lang (push) Failing after 0s
CodeQL Security Analysis / CodeQL Analysis (push) Failing after 0s
PR CI / pr-ci (push) Failing after 1s
Slow tests on important models (on Push - A10) / Get all modified files (push) Failing after 1s
Slow tests on important models (on Push - A10) / Model CI (push) Has been skipped
Self-hosted runner (benchmark) / Benchmark (aws-g5-4xlarge-cache) (push) Has been cancelled
New model PR merged notification / Notify new model (push) Has been cancelled
Update Transformers metadata / build_and_package (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 11:57:37 +08:00

3.5 KiB

Server optimizations

transformers serve includes optimizations to improve throughput and reduce memory usage.

Continuous batching

Continuous batching dynamically groups and interleaves requests to share forward passes on the GPU. New requests join the batch as others progress through prefill. Completed requests drop out after decoding. This increases GPU utilization and throughput without compromising latency.

Add --continuous-batching to enable it.

transformers serve \
  --continuous-batching \
  --attn-implementation "sdpa"

Quantization

Quantization reduces memory usage by mapping weights to a lower precision. transformers serve is compatible with all quantization methods in Transformers. It supports pre-quantized models and runtime quantization.

Pre-quantized models don't require any changes. They offer the best balance between performance and accuracy. Install the appropriate quantization library. Then pass the pre-quantized model from the Hub to the model argument.

curl http://localhost:8000/v1/responses \
  -H "Content-Type: application/json" \
  -d '{
    "model": "Qwen/Qwen3-8B-GGUF",
    "stream": true,
    "input": "Tell me a three sentence bedtime story about a unicorn."
  }'

Use --quantization to quantize a model at runtime. This is useful for new checkpoints or finetunes without pre-quantized weights. Only bitsandbytes 4-bit and 8-bit quantization are supported.

transformers serve \
  --quantization bnb-4bit

Attention backend

An optimized attention backend improves memory efficiency and speeds up inference.

transformers serve \
  --continuous-batching \
  --attn-implementation "flash_attention_2"

Apple Silicon (Metal flash attention)

Install kernels to make transformers serve default to kernels-community/metal-flash-sdpa on MPS. The Metal flash kernel runs 1.66x faster than SDPA with generate_batch on 100 samples of gsm8k, Qwen2.5-0.5B-Instruct and MPS fp16. It matches SDPA token-for-token under greedy decoding.

pip install kernels
transformers serve

A warning prints at startup confirming the auto-selection. Pass --attn-implementation sdpa to opt out.

transformers serve --attn-implementation sdpa

Compile

torch.compile traces and compiles the decode loop for faster inference.

Note

Compile is incompatible with continuous batching.

transformers serve \
  --compile

Data type

The "bfloat16" or "float16" data types save memory and increase throughput.

transformers serve \
  --continuous-batching \
  --dtype "bfloat16"