vllm/vllm
youkaichao 3cd9b5bb2d
[Core][Distributed] use existing torch.cuda.device (#4318)
[Core][Distributed] use existing torch.cuda.device context manager (#4318)
2024-04-24 09:00:20 -07:00
..
attention [Mypy] Part 3 fix typing for nested directories for most of directory (#4161) 2024-04-22 21:32:44 -07:00
core [Core] Scheduling optimization 2 (#4280) 2024-04-23 08:02:11 +00:00
distributed [Core][Distributed] use existing torch.cuda.device (#4318) 2024-04-24 09:00:20 -07:00
engine [Speculative decoding 7/9] Speculative decoding end-to-end correctness tests. (#3951) 2024-04-23 08:02:36 +00:00
entrypoints [Bugfix][Frontend] Raise exception when file-like chat template fails to be opened (#4292) 2024-04-23 18:19:03 +00:00
executor [Bugfix] Add init_cached_hf_modules to RayWorkerWrapper (#4286) 2024-04-23 09:28:35 -07:00
lora [Mypy] Part 3 fix typing for nested directories for most of directory (#4161) 2024-04-22 21:32:44 -07:00
model_executor [BUG] fixed fp8 conflict with aqlm (#4307) 2024-04-23 18:26:33 -07:00
spec_decode [Speculative decoding 7/9] Speculative decoding end-to-end correctness tests. (#3951) 2024-04-23 08:02:36 +00:00
transformers_utils [Mypy] Part 3 fix typing for nested directories for most of directory (#4161) 2024-04-22 21:32:44 -07:00
usage [mypy] Add mypy type annotation part 1 (#4006) 2024-04-12 14:35:50 -07:00
worker [Core][Distributed] use cpu/gloo to initialize pynccl (#4248) 2024-04-23 18:32:19 -07:00
__init__.py Bump version of 0.4.1 (#4177) 2024-04-19 01:00:22 -07:00
_custom_ops.py [Kernel] FP8 support for MoE kernel / Mixtral (#4244) 2024-04-24 01:18:23 +00:00
block.py Add Automatic Prefix Caching (#2762) 2024-03-02 00:50:01 -08:00
config.py [Speculative decoding 7/9] Speculative decoding end-to-end correctness tests. (#3951) 2024-04-23 08:02:36 +00:00
logger.py [Core][Logging] Add last frame information for better debugging (#4278) 2024-04-23 09:45:52 -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 Support eos_token_id from generation_config.json (#4182) 2024-04-19 04:13:36 +00:00
sequence.py [Core] Scheduling optimization 2 (#4280) 2024-04-23 08:02:11 +00:00
test_utils.py [Core][Refactor] move parallel_utils into vllm/distributed (#3950) 2024-04-10 15:33:30 -07:00
utils.py [Core][Distributed] use absolute path for library file (#4271) 2024-04-22 17:21:48 -07:00