v0.2.6
vllm-project/vllmv0.2.6Dec 17, 2023by github-actions[bot]
AI Summary
This release introduces fast model execution capabilities using CUDA/HIP graphs and adds support for W4A16 GPTQ quantization, alongside several bug fixes for Mixtral weight loading and memory profiling.
Key Highlights
- Fast model execution with CUDA/HIP graph
- W4A16 GPTQ support
- Fix memory profiling with tensor parallelism
- Fix *.bin weight loading for Mixtral models
- Added Phi 2 to supported models
New Features
- Added GPTQ support
- Optimized model execution with CUDA graph
- Remove dependency on CuPy
- Temporarily enforce eager mode for GPTQ models
Full Release Notes
## Major changes * Fast model execution with CUDA/HIP graph * W4A16 GPTQ support (thanks to @chu-tianxiang) * Fix memory profiling with tensor parallelism * Fix *.bin weight loading for Mixtral models ## What's Changed * Fix typing in generate function for AsyncLLMEngine & add toml to requirements-dev by @mezuzza in https://github.com/vllm-project/vllm/pull/2100 * Fix Dockerfile.rocm by @tjtanaa in https://github.com/vllm-project/vllm/pull/2101 * avoid multiple redefinition by @MitchellX in https://github.com/vllm-project/vllm/pull/1817 * Add a flag to include stop string in output text by @yunfeng-scale in https://github.com/vllm-project/vllm/pull/1976 * Add GPTQ support by @chu-tianxiang in https://github.com/vllm-project/vllm/pull/916 * [Docs] Add quantization support to docs by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2135 * [ROCm] Temporarily remove GPTQ ROCm support by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2138 * simplify loading weights logic by @esmeetu in https://github.com/vllm-project/vllm/pull/2133 * Optimize model execution with CUDA graph by @WoosukKwon in https://github.com/vllm-project/vllm/pull/1926 * [Minor] Delete Llama tokenizer warnings by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2146 * Fix all-reduce memory usage by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2151 * Pin PyTorch & xformers versions by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2155 * Remove dependency on CuPy by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2152 * [Docs] Add CUDA graph support to docs by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2148 * Temporarily enforce eager mode for GPTQ models by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2154 * [Minor] Add more detailed explanation on `quantization` argument by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2145 * [Minor] Fix xformers version by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2158 * [Minor] Add Phi 2 to supported models by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2159 * Make sampler less blocking by @Yard1 in https://github.com/vllm-project/vllm/pull/1889 * [Minor] Fix a typo in .pt weight support by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2160 * Disable CUDA graph for SqueezeLLM by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2161 * Bump up to v0.2.6 by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2157 ## New Contributors * @mezuzza made their first contribution in https://github.com/vllm-project/vllm/pull/2100 * @MitchellX made their first contribution in https://github.com/vllm-project/vllm/pull/1817 **Full Changelog**: https://github.com/vllm-project/vllm/compare/v0.2.5...v0.2.6