Commit Graph

9263 Commits

Author SHA1 Message Date
Zhuohan Li
5020e1e80c Non-streaming simple fastapi server (#144) 2023-06-10 10:43:07 -07:00
Zhuohan Li
4298374265 Add docstrings for LLMServer and related classes and examples (#142) 2023-06-07 18:25:20 +08:00
Woosuk Kwon
e38074b1e6 Support FP32 (#141) 2023-06-07 00:40:21 -07:00
Woosuk Kwon
376725ce74 [PyPI] Packaging for PyPI distribution (#140) 2023-06-05 20:03:14 -07:00
Woosuk Kwon
456941cfe4 [Docs] Write the Adding a New Model section (#138) 2023-06-05 20:01:26 -07:00
Zhuohan Li
1a956e136b Fix various issues of async servers (#135) 2023-06-05 23:44:50 +08:00
Woosuk Kwon
8274ca23ac Add docstrings for LLM (#137) 2023-06-04 12:52:41 -07:00
Woosuk Kwon
62ec38ea41 Document supported models (#127) 2023-06-02 22:35:17 -07:00
Woosuk Kwon
0eda2e0953 Add .readthedocs.yaml (#136) 2023-06-02 22:27:44 -07:00
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