3 people 1356df53bd [Kernel] Use flash-attn for decoding (#3648) ...

Co-authored-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
Co-authored-by: LiuXiaoxuanPKU <lilyliupku@gmail.com>

2024-05-13 15:50:33 -07:00

..

attention

[Kernel] Use flash-attn for decoding (#3648)

2024-05-13 15:50:33 -07:00

core

[Scheduler] Warning upon preemption and Swapping (#4647)

2024-05-13 23:50:44 +09:00

distributed

[Core][Distributed] refactor custom allreduce to support multiple tp groups (#4754)

2024-05-12 17:47:59 -07:00

engine

[Frontend] [Core] perf: Automatically detect vLLM-tensorized model, update tensorizer to version 2.9.0 (#4208)

2024-05-13 14:57:07 -07:00

entrypoints

[Model][Misc] Add e5-mistral-7b-instruct and Embedding API (#3734)

2024-05-11 11:30:37 -07:00

executor

[Model][Misc] Add e5-mistral-7b-instruct and Embedding API (#3734)

2024-05-11 11:30:37 -07:00

logging

[MISC] Rework logger to enable pythonic custom logging configuration to be provided (#4273)

2024-05-01 17:34:40 -07:00

lora

[Core] Faster startup for LoRA enabled models (#4634)

2024-05-08 10:33:18 -07:00

model_executor

[Frontend] [Core] perf: Automatically detect vLLM-tensorized model, update tensorizer to version 2.9.0 (#4208)

2024-05-13 14:57:07 -07:00

spec_decode

[Speculative decoding] Improve n-gram efficiency (#4724)

2024-05-13 15:00:13 -07:00

transformers_utils

[Model] Snowflake arctic model implementation (#4652)

2024-05-09 22:37:14 +00:00

usage

[Misc] centralize all usage of environment variables (#4548)

2024-05-02 11:13:25 -07:00

worker

[Kernel] Use flash-attn for decoding (#3648)

2024-05-13 15:50:33 -07:00

__init__.py

[Model][Misc] Add e5-mistral-7b-instruct and Embedding API (#3734)

2024-05-11 11:30:37 -07:00

_custom_ops.py

[Core]fix type annotation for swap_blocks (#4726)

2024-05-10 21:52:48 +09:00

block.py

Add Automatic Prefix Caching (#2762)

2024-03-02 00:50:01 -08:00

config.py

[Speculative decoding] Improve n-gram efficiency (#4724)

2024-05-13 15:00:13 -07:00

envs.py

[Frontend] [Core] perf: Automatically detect vLLM-tensorized model, update tensorizer to version 2.9.0 (#4208)

2024-05-13 14:57:07 -07:00

logger.py

[Misc] centralize all usage of environment variables (#4548)

2024-05-02 11:13:25 -07:00

outputs.py

[Model][Misc] Add e5-mistral-7b-instruct and Embedding API (#3734)

2024-05-11 11:30:37 -07:00

pooling_params.py

[Model][Misc] Add e5-mistral-7b-instruct and Embedding API (#3734)

2024-05-11 11:30:37 -07:00

py.typed

Add py.typed so consumers of vLLM can get type checking (#1509)

2023-10-30 14:50:47 -07:00

sampling_params.py

[Bugfix] Use random seed if seed is -1 (#4531)

2024-05-01 10:41:17 -07:00

sequence.py

[Model][Misc] Add e5-mistral-7b-instruct and Embedding API (#3734)

2024-05-11 11:30:37 -07:00

utils.py

[Kernel] Refactor FP8 kv-cache with NVIDIA float8_e4m3 support (#4535)

2024-05-09 18:04:17 -06:00