Woosuk Kwon 5cb213c85e Add flash-attn op

2024-04-17 18:02:28 +00:00

..

attention

Remove

2024-04-16 08:05:36 +00:00

core

[Typing] Fix Sequence type GenericAlias only available after Python 3.9. (#4092)

2024-04-15 14:47:31 -07:00

distributed

[Core] avoid too many cuda context by caching p2p test (#4021)

2024-04-13 23:40:21 -07:00

engine

Merge branch 'main' into woosuk-tpu

2024-04-16 07:56:53 +00:00

entrypoints

LM Format Enforcer Guided Decoding Support (#3868)

2024-04-16 05:54:57 +00:00

executor

Minor

2024-04-16 17:04:46 +00:00

lora

[Bugfix] Fix LoRA bug (#4032)

2024-04-12 16:56:37 -07:00

model_executor

Add flash-attn op

2024-04-17 18:02:28 +00:00

spec_decode

[Misc] [Core] Implement RFC "Augment BaseExecutor interfaces to enable hardware-agnostic speculative decoding" (#3837)

2024-04-09 11:44:15 -07:00

transformers_utils

[mypy] Add mypy type annotation part 1 (#4006)

2024-04-12 14:35:50 -07:00

usage

[mypy] Add mypy type annotation part 1 (#4006)

2024-04-12 14:35:50 -07:00

worker

JAX-based TPU worker

2024-04-16 17:37:11 +00:00

__init__.py

[Core] enable out-of-tree model register (#3871)

2024-04-06 17:11:41 -07:00

_custom_ops.py

[Misc] Add indirection layer for custom ops (#3913)

2024-04-10 20:26:07 -07:00

block.py

Add Automatic Prefix Caching (#2762)

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

config.py

Minor

2024-04-16 09:41:53 +00:00

logger.py

[Doc] Add typing hints / mypy types cleanup (#3816)

2024-04-11 17:17:21 -07:00

outputs.py

[BugFix] Fix handling of stop strings and stop token ids (#3672)

2024-04-11 15:34:12 -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

[mypy] Add mypy type annotation part 1 (#4006)

2024-04-12 14:35:50 -07:00

sequence.py

[mypy] Add mypy type annotation part 1 (#4006)

2024-04-12 14:35:50 -07:00

test_utils.py

[Core][Refactor] move parallel_utils into vllm/distributed (#3950)

2024-04-10 15:33:30 -07:00

utils.py

Merge branch 'main' into woosuk-tpu

2024-04-16 07:56:53 +00:00