MODEL_NAME | facebook/opt-125m | str | Hugging Face repo ID or local filesystem path for the model weights. Change this to deploy a different model. |
MODEL_REVISION | main | str | The model revision to load. |
TOKENIZER | None | str | The name or path of the Hugging Face tokenizer to use. |
SKIP_TOKENIZER_INIT | False | bool | If True, skips the initialization of the tokenizer and detokenizer. |
TOKENIZER_MODE | auto | auto, slow | The tokenizer mode. |
TRUST_REMOTE_CODE | False | bool | If True, trusts remote code from Hugging Face. |
DOWNLOAD_DIR | None | str | The directory to download and load the model weights from. |
LOAD_FORMAT | auto | str | The format of the model weights to load. |
HF_TOKEN | - | str | Hugging Face token required to download gated/private models. Not needed for public models. Provide it via secrets. |
DTYPE | auto | auto, half, float16, bfloat16, float, float32 | The data type for model weights and activations. |
KV_CACHE_DTYPE | auto | auto, fp8 | The data type for KV cache storage. |
QUANTIZATION_PARAM_PATH | None | str | The path to the JSON file containing the KV cache scaling factors. |
MAX_MODEL_LEN | None | int | Maximum context length (tokens) the engine will allocate KV cache for. Lower it to reduce VRAM usage; raise it for long-context models if supported and you have VRAM. |
GUIDED_DECODING_BACKEND | outlines | outlines, lm-format-enforcer | The default engine for guided decoding. |
DISTRIBUTED_EXECUTOR_BACKEND | None | ray, mp | The backend to use for distributed serving. |
WORKER_USE_RAY | False | bool | Deprecated. Use DISTRIBUTED_EXECUTOR_BACKEND=ray instead. |
PIPELINE_PARALLEL_SIZE | 1 | int | The number of pipeline stages. |
TENSOR_PARALLEL_SIZE | 1 | int | Tensor parallel degree (number of GPUs to shard across). On multi-GPU machines, this worker auto-sets it to the number of visible GPUs. |
MAX_PARALLEL_LOADING_WORKERS | None | int | Load model sequentially in multiple batches to avoid RAM OOM when using tensor parallelism and large models. |
RAY_WORKERS_USE_NSIGHT | False | bool | If True, uses nsight to profile Ray workers. |
ENABLE_PREFIX_CACHING | False | bool | If True, enables automatic prefix caching. |
DISABLE_SLIDING_WINDOW | False | bool | If True, disables the sliding window, capping to the sliding window size. |
USE_V2_BLOCK_MANAGER | False | bool | If True, uses the BlockSpaceMangerV2. |
NUM_LOOKAHEAD_SLOTS | 0 | int | The number of lookahead slots, an experimental scheduling configuration for speculative decoding. |
SEED | 0 | int | The random seed for operations. |
NUM_GPU_BLOCKS_OVERRIDE | None | int | If specified, this value overrides the GPU profiling result for the number of GPU blocks. |
MAX_NUM_BATCHED_TOKENS | None | int | The maximum number of batched tokens per iteration. |
MAX_NUM_SEQS | 256 | int | Upper bound on sequences batched per iteration (affects throughput, VRAM, and tail latency). Higher can improve throughput for many concurrent short requests; lower reduces VRAM usage. |
MAX_LOGPROBS | 20 | int | The maximum number of log probabilities to return when logprobs is specified in SamplingParams. |
DISABLE_LOG_STATS | False | bool | If True, disables logging statistics. |
QUANTIZATION | None | awq, squeezellm, gptq, bitsandbytes | Quantization backend for loading quantized checkpoints (AWQ/GPTQ/…) or BitsAndBytes. Must match the checkpoint format. |
ROPE_SCALING | None | dict | The RoPE scaling configuration in JSON format. |
ROPE_THETA | None | float | The RoPE theta value. Use with ROPE_SCALING. |
TOKENIZER_POOL_SIZE | 0 | int | The size of the tokenizer pool for asynchronous tokenization. |
TOKENIZER_POOL_TYPE | ray | str | The type of the tokenizer pool for asynchronous tokenization. |
TOKENIZER_POOL_EXTRA_CONFIG | None | dict | Extra configuration for the tokenizer pool. |