vLLM
/
vllm
mirror of https://github.com/vllm-project/vllm.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
7,204 Commits 73 Branches 67 Tags 518 MiB
Commit Graph

7204 Commits

Author SHA1 Message Date
Woosuk Kwon 211318d44a
Add throughput benchmarking script (#133) 2023-05-28 03:20:05 -07:00
Woosuk Kwon 337871c6fd
Enable LLaMA fast tokenizer (#132) 2023-05-28 02:51:42 -07:00
Woosuk Kwon 56b7f0efa4
Add a doc for installation (#128) 2023-05-27 01:13:06 -07:00
Woosuk Kwon d721168449
Improve setup script & Add a guard for bfloat16 kernels (#130) 2023-05-27 00:59:32 -07:00
Woosuk Kwon 4a151dd453
Add activation registry (#126) 2023-05-25 00:09:07 -07:00
Zhuohan Li 057daef778
OpenAI Compatible Frontend (#116) 2023-05-23 21:39:50 -07:00
Woosuk Kwon e86717833d
Incrementally decode output tokens (#121) 2023-05-23 20:46:32 -07:00
Woosuk Kwon aedba6d5ec
Print warnings/errors for large swap space (#123) 2023-05-23 18:22:26 -07:00
Woosuk Kwon a283ec2eec
Add contributing guideline and mypy config (#122) 2023-05-23 17:58:51 -07:00
Woosuk Kwon 3f942acfe1
Fix latency benchmark script (#118) 2023-05-22 17:03:40 -07:00
Woosuk Kwon 19d2899439
Add initial sphinx docs (#120) 2023-05-22 17:02:44 -07:00
Woosuk Kwon 655a5e48df
Introduce LLM class for offline inference (#115) 2023-05-21 17:04:18 -07:00
Woosuk Kwon f746ced08d
Implement stop strings and best_of (#114) 2023-05-21 11:18:00 -07:00
Woosuk Kwon c3442c1f6f
Refactor system architecture (#109) 2023-05-20 13:06:59 -07:00
Zhuohan Li 7297fa6f7c
Remove unused parts in Megatron-LM code and add copyright notice (#110) 2023-05-20 09:11:34 -06:00
Zhuohan Li b7955ef17b
Fix timeout error in the FastAPI frontend (#34) 2023-05-19 14:00:46 -06:00
Zhuohan Li f756799b84
Use runtime profiling to replace manual memory analyzers (#81) 2023-05-19 11:35:44 -06:00
Woosuk Kwon 825d8892b5
Use pytest format for unit tests (#107) 2023-05-17 17:11:23 -07:00
Woosuk Kwon b322fd1607
Add docstrings to some modules and classes (#100) 2023-05-14 22:32:38 -07:00
Woosuk Kwon 667ba3995c
Add copyright headers to source files adapted from FT (#104) 2023-05-14 22:19:19 -07:00
Woosuk Kwon 707ec647bb
Add copyright headers for HF models (#103) 2023-05-14 21:54:32 -07:00
Woosuk Kwon 89988ec8c2
Add Apache-2.0 license (#102) 2023-05-14 18:05:19 -07:00
Woosuk Kwon 6208d622ca
Minor code cleaning for SamplingParams (#99) 2023-05-12 18:07:09 -07:00
Woosuk Kwon 42f1042e1c
Enhance SamplingParams (#96) 2023-05-11 15:45:30 -07:00
Woosuk Kwon 55f8b0a5de
Implement presence and frequency penalties (#95) 2023-05-10 23:39:12 -07:00
Woosuk Kwon 9f88db35da
Support top-k sampling (#94) 2023-05-10 12:51:36 -07:00
Woosuk Kwon ae356774ab
Avoid sorting waiting queue & Minor code cleaning (#93) 2023-05-10 01:57:07 -07:00
Woosuk Kwon e331957784
Log system stats (#90) 2023-05-10 01:06:53 -07:00
Woosuk Kwon 8d66a7b6d7
Rename variables and methods (#91) 2023-05-10 00:58:31 -07:00
Woosuk Kwon ce26e57fd3
Update sample prompts in `simple_server.py` (#89) 2023-05-09 16:47:39 -07:00
Woosuk Kwon 85eb631839
Use slow tokenizer for LLaMA (#84) 2023-05-09 16:03:44 -07:00
Woosuk Kwon add055e151
Enhance model loader (#83) 2023-05-09 15:46:42 -07:00
Woosuk Kwon 7c041ab578
Refactor system architecture (#82) 2023-05-09 15:30:12 -07:00
Woosuk Kwon 8917782af6
Add a system logger (#85) 2023-05-08 23:03:35 -07:00
Woosuk Kwon 7addca5935
Specify python package dependencies in requirements.txt (#78) 2023-05-07 16:30:43 -07:00
Woosuk Kwon c84e924287
[Minor] Fix a dtype bug (#79) 2023-05-06 02:12:12 -07:00
Woosuk Kwon c9d5b6d4a8
Replace FlashAttention with xformers (#70) 2023-05-05 02:01:08 -07:00
Woosuk Kwon 189ae23133
Use dtype from model config & Add Dolly V2 (#63) 2023-05-04 03:05:37 -07:00
Woosuk Kwon e548c1488a
Add support for GPT-2 (#60) 2023-05-04 02:59:56 -07:00
Woosuk Kwon 130d5fd8c7
Fix a bug in attention kernel (#68) 2023-05-04 02:56:09 -07:00
Woosuk Kwon e070829ae8
Support bfloat16 data type (#54) 2023-05-03 14:09:44 -07:00
Woosuk Kwon 436e523bf1
Refactor attention kernels (#53) 2023-05-03 13:40:13 -07:00
Zhuohan Li 27f1410d06
New weight loader without np copy (#52) 2023-05-03 15:32:04 +08:00
Zhuohan Li 4858f3bb45
Add an option to launch cacheflow without ray (#51) 2023-04-30 15:42:17 +08:00
Woosuk Kwon a96d63c21d
Add support for GPT-NeoX (Pythia) (#50) 2023-04-28 00:32:10 -07:00
Woosuk Kwon aa50b17ca7 Change plotting script 2023-04-17 04:49:14 +00:00
Woosuk Kwon 0f4b32199e
Support various block sizes & Change default block size to 16 (#38) 2023-04-15 09:03:24 -07:00
Woosuk Kwon 84eee24e20
Collect system stats in scheduler & Add scripts for experiments (#30) 2023-04-12 15:03:49 -07:00
Siyuan (Ryans) Zhuang e3cec88aa5
Memcpy kernel for flash attention (#29) 
* optimize

* add benchmark

* add assert

* add test
 2023-04-10 18:22:49 -07:00
Woosuk Kwon b9926f7f66
Support block size 32 (#35) 2023-04-09 23:07:18 -07:00