v0.2.5
vllm-project/vllmv0.2.5Dec 14, 2023by github-actions[bot]
AI Summary
Performance optimization for Mixtral models via expert parallelism and fixes for long context input positions are the main highlights, alongside a new lazy model loader and PyTorch upgrade.
Key Highlights
- Optimize Mixtral performance with expert parallelism
- Fix input positions for long context with sliding window
- Implement lazy model loader
- Upgrade PyTorch version to v2.1.1
New Features
- Remove python 3.10 requirement
- Upgrade transformers version to 4.36.0
- gqa added to mpt attn
- ROCm xformers upgrade
Full Release Notes
## Major changes * Optimize Mixtral performance with expert parallelism (thanks to @Yard1) * [BugFix] Fix input positions for long context with sliding window ## What's Changed * Update Dockerfile to support Mixtral by @simon-mo in https://github.com/vllm-project/vllm/pull/2027 * Remove python 3.10 requirement by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2040 * [CI/CD] Upgrade PyTorch version to v2.1.1 by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2045 * Upgrade transformers version to 4.36.0 by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2046 * Remove einops from dependencies by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2049 * gqa added to mpt attn by @megha95 in https://github.com/vllm-project/vllm/pull/1938 * Update Dockerfile to build Megablocks by @simon-mo in https://github.com/vllm-project/vllm/pull/2042 * Fix peak memory profiling by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2031 * Implement lazy model loader by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2044 * [ROCm] Upgrade xformers version dependency for ROCm; update documentations by @tjtanaa in https://github.com/vllm-project/vllm/pull/2079 * Update installation instruction for CUDA 11.8 by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2086 * [Docs] Add notes on ROCm-supported models by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2087 * [BugFix] Fix input positions for long context with sliding window by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2088 * Mixtral expert parallelism by @Yard1 in https://github.com/vllm-project/vllm/pull/2090 * Bump up to v0.2.5 by @WoosukKwon in https://github.com/vllm-project/vllm/pull/2095 **Full Changelog**: https://github.com/vllm-project/vllm/compare/v0.2.4...v0.2.5